Searched defs:on_value (Results 1 – 8 of 8) sorted by relevance
35 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
53 const Tensor& on_value = ctx->input(2); in Compute() local
53 const TfLiteTensor* on_value; member73 const T on_value = *GetTensorData<T>(op_context.on_value); in OneHotComputeImpl() local
33 TensorType dtype, int axis = -1, T on_value = 1, in OneHotOpModel()
86 const xla::XlaOp& indices, const xla::XlaOp& on_value, in OneHot()
99 float on_value = 1; field
2905 on_value=None, argument
2539 on_value=1.0, argument