Home
last modified time | relevance | path

Searched refs:CheckTensorBounds (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc27 using tensorforest::CheckTensorBounds;
60 if (!CheckTensorBounds(context, input_tensor)) return; in Compute()
61 if (!CheckTensorBounds(context, indices_tensor)) return; in Compute()
62 if (!CheckTensorBounds(context, deltas_tensor)) return; in Compute()
Dreinterpret_string_to_float_op.cc31 using tensorforest::CheckTensorBounds;
57 if (!CheckTensorBounds(context, input_data)) return; in Compute()
Dtree_utils.h284 inline bool CheckTensorBounds(OpKernelContext* context, const Tensor& tensor) { in CheckTensorBounds() function
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Drouting_function_op.cc43 using tensorforest::CheckTensorBounds;
99 if (!CheckTensorBounds(context, input_data)) return; in Compute()
Dk_feature_routing_function_op.cc43 using tensorforest::CheckTensorBounds;
114 if (!CheckTensorBounds(context, input_data)) return; in Compute()
Dhard_routing_function_op.cc43 using tensorforest::CheckTensorBounds;
108 if (!CheckTensorBounds(context, input_data)) return; in Compute()
Dstochastic_hard_routing_function_op.cc47 using tensorforest::CheckTensorBounds;
119 if (!CheckTensorBounds(context, input_data)) return; in Compute()