Home
last modified time | relevance | path

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

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc62 TfLiteTensor& input_limits = in EvalT() local
68 const bool broadcast_limits = NumElements(&input_limits) == 1; in EvalT()
75 if (!broadcast_limits) in_sizes.push_back(input_limits.dims->data[0]); in EvalT()
89 const T* limits = GetTensorData<T>(&input_limits); in EvalT()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc590 DimensionVector input_limits; in HandleConvolution() local
595 input_limits.push_back(1); in HandleConvolution()
615 input_limits.push_back(input_limit); in HandleConvolution()
628 if (input_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
630 input_limits[spatial_dimension] == window_dim.base_dilation() && in HandleConvolution()
632 std::max<int64>(1, input_limits[spatial_dimension] - 1) == in HandleConvolution()
635 valid_position_counts.push_back(input_limits[spatial_dimension]); in HandleConvolution()
639 if (input_limits[spatial_dimension] == 1 && in HandleConvolution()
676 lhs_spatial_index >= input_limits[spatial_dimension]) { in HandleConvolution()