Searched refs:strides_shape (Results 1 – 4 of 4) sorted by relevance
| /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 | 33 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel() argument
46 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1684 std::initializer_list<int> strides_shape, in StridedSliceOpModel() argument
1691 strides_ = AddConstInput(TensorType_INT32, strides_data, strides_shape); in StridedSliceOpModel()
1698 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1567 ShapeHandle begin_shape, end_shape, strides_shape; in __anondb9326b22502() local
1570 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &strides_shape)); in __anondb9326b22502()
1572 TF_RETURN_IF_ERROR(c->Merge(begin_shape, strides_shape, &begin_shape)); in __anondb9326b22502()
|