Home
last modified time | relevance | path

Searched refs:pad_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dpad_test.cc98 TfLiteIntArray* pad_dims = IntArrayFromInts(pad_dims_data); in TestPadFloat() local
105 CreateTensor(pad_data, pad_dims), in TestPadFloat()
123 TfLiteIntArray* pad_dims = IntArrayFromInts(pad_dims_data); in TestPadV2Float() local
132 CreateTensor(input_data, input_dims), CreateTensor(pad_data, pad_dims), in TestPadV2Float()
153 TfLiteIntArray* pad_dims = IntArrayFromInts(pad_dims_data); in TestPadQuantized() local
162 CreateTensor(pad_data, pad_dims), in TestPadQuantized()
189 TfLiteIntArray* pad_dims = IntArrayFromInts(pad_dims_data); in TestPadV2Quantized() local
201 CreateTensor(pad_data, pad_dims), in TestPadV2Quantized()
229 const int pad_dims[] = {2, 4, 2}; in TF_LITE_MICRO_TEST() local
236 tflite::testing::TestPadFloat(input_dims, input_values, pad_dims, pad_values, in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_with_default_op.py155 pad_dims = params_shape.partitioned_dim_sizes[:-1] + (
158 pad_dims, [])
167 pad_dims = None
169 pad_dims = (constant_op.constant(1, dtype=row_splits_dtype),) + (
178 pad_dims = batch_dimensions + (
183 pad_dims, params_shape.inner_dim_sizes)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2935 std::vector<int> pad_dims; in ConvertStridedSliceHelper() local
2938 pad_dims.push_back(i); in ConvertStridedSliceHelper()
2941 if (pad_dims.empty()) { in ConvertStridedSliceHelper()
2954 } else if (pad_dims.size() == 1) { in ConvertStridedSliceHelper()
2958 if (pad_dims[0] != 2) { in ConvertStridedSliceHelper()
2959 pad_dims.push_back(2); in ConvertStridedSliceHelper()
2961 pad_dims.push_back(3); in ConvertStridedSliceHelper()
2963 } else if (pad_dims.size() > 2) { in ConvertStridedSliceHelper()
2969 std::sort(pad_dims.begin(), pad_dims.end()); in ConvertStridedSliceHelper()
2974 for (int i = 0; i < pad_dims.size(); i++) { in ConvertStridedSliceHelper()
[all …]
Dconvert_nodes_test.cc7119 std::vector<int> pad_dims; in TEST_P() member
7272 AddTestWeights<int32>("padding", p.pad_dims, p.pad_values); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3394 auto pad_dims = padding_dimensions.size(); in HandlePad() local
3395 if (pad_dims < dimension_index && in HandlePad()
3398 pad->operand(0)->operand(0)->shape().rank() <= pad_dims) { in HandlePad()
3407 for (int i = 0; i < pad_dims; ++i) { in HandlePad()