Home
last modified time | relevance | path

Searched defs:off_value (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Done_hot_op.h36 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
Done_hot_op.cc54 const Tensor& off_value = ctx->input(3); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/
Done_hot.cc54 const TfLiteTensor* off_value; member
74 const T off_value = *GetTensorData<T>(op_context.off_value); in OneHotComputeImpl() local
Done_hot_test.cc34 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc87 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { in OneHot()
/external/tensorflow/tensorflow/contrib/tensor_forest/proto/
Dtensor_forest_params.proto100 float off_value = 2; field
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2906 off_value=None, argument
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py2540 off_value=0.0, argument