Searched defs:off_value (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Done_hot_op.h36 const typename TTypes<T>::ConstScalar& off_value)
37 : indices_(indices), on_value_(on_value), off_value_(off_value) {}
61 const typename TTypes<T>::ConstScalar& off_value,
63 generator::OneGenerator<T, TI> generator(indices, on_value, off_value);
34 OneGenerator(const typename TTypes<TI>::ConstMatrix& indices, const typename TTypes<T>::ConstScalar& on_value, const typename TTypes<T>::ConstScalar& off_value) argument
58 Compute( const Device& d, const typename TTypes<TI>::ConstMatrix& indices, const typename TTypes<T>::ConstScalar& on_value, const typename TTypes<T>::ConstScalar& off_value, typename TTypes<T, 3>::Tensor* output) argument
H A Done_hot_op.cc54 const Tensor& off_value = ctx->input(3); variable
71 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value.shape()),
72 errors::InvalidArgument("off_value must be a scalar, but got: ",
73 off_value.shape().DebugString()));
93 auto off_value_t = off_value.scalar<T>();
152 const typename TTypes<T>::ConstScalar& off_value, \
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_helpers.cc231 const xla::ComputationDataHandle& off_value,
269 // Selects the user-provided off_value and on_value values.
272 builder->Broadcast(off_value, output_shape.dim_sizes()));
226 OneHot(xla::ComputationBuilder* builder, int64 depth, int axis, DataType index_type, const TensorShape& indices_shape, const xla::ComputationDataHandle& indices, const xla::ComputationDataHandle& on_value, const xla::ComputationDataHandle& off_value, xla::ComputationDataHandle* one_hot) argument

Completed in 768 milliseconds