Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dragged_range_op.cc48 bool broadcast_starts = starts_in.shape().dims() == 0; in Compute() local
55 if (!broadcast_starts) in_sizes.push_back(starts_in.shape().dim_size(0)); in Compute()
77 T start = broadcast_starts ? starts(0) : starts(row); in Compute()
94 T value = broadcast_starts ? starts(0) : starts(row); in Compute()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc67 const bool broadcast_starts = NumElements(&input_starts) == 1; in EvalT() local
74 if (!broadcast_starts) in_sizes.push_back(input_starts.dims->data[0]); in EvalT()
102 const T start = broadcast_starts ? starts[0] : starts[row]; in EvalT()
124 T value = broadcast_starts ? starts[0] : starts[row]; in EvalT()