Home
last modified time | relevance | path

Searched refs:broadcast_limits (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_range_op.cc49 bool broadcast_limits = limits_in.shape().dims() == 0; in Compute() local
56 if (!broadcast_limits) in_sizes.push_back(limits_in.shape().dim_size(0)); in Compute()
78 T limit = broadcast_limits ? limits(0) : limits(row); in Compute()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc68 const bool broadcast_limits = NumElements(&input_limits) == 1; in EvalT() local
75 if (!broadcast_limits) in_sizes.push_back(input_limits.dims->data[0]); in EvalT()
103 const T limit = broadcast_limits ? limits[0] : limits[row]; in EvalT()