Searched defs:off_value (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | one_hot_op.h | 36 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 D | one_hot_op.cc | 54 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 D | xla_helpers.cc | 231 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