Searched refs:strides_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 60 const int* end_shape, const int* strides_shape, in TestStridedSliceFloat() argument
70 TfLiteIntArray* strides_dims = IntArrayFromInts(strides_shape); in TestStridedSliceFloat()
92 const int* strides_shape, TfLiteStridedSliceParams* builtin_data, in TestStridedSliceQuantized() argument
99 TfLiteIntArray* strides_dims = IntArrayFromInts(strides_shape); in TestStridedSliceQuantized()
130 const int strides_shape[] = {1, 5}; in TF_LITE_MICRO_TEST() local
142 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
151 const int strides_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
163 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
172 const int strides_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
184 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | flops_registry.py | 405 strides_shape = list(node.attr["strides"].list.i)
406 strides_product = strides_shape[1] * strides_shape[2]
428 strides_shape = list(node.attr["strides"].list.i)
429 strides_product = strides_shape[1] * strides_shape[2]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 36 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel() argument
49 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1706 ShapeHandle begin_shape, end_shape, strides_shape; in __anondb9326b22502() local
1709 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &strides_shape)); in __anondb9326b22502()
1711 TF_RETURN_IF_ERROR(c->Merge(begin_shape, strides_shape, &begin_shape)); in __anondb9326b22502()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2515 std::initializer_list<int> strides_shape, in StridedSliceOpModel() argument
2522 strides_ = AddConstInput(TensorType_INT32, strides_data, strides_shape); in StridedSliceOpModel()
2530 {input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|