Searched refs:begin_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | slice_test.cc | 30 std::initializer_list<int> begin_shape, in SliceOpModel() argument
39 BuildInterpreter({input_shape, begin_shape, size_shape}); in SliceOpModel()
|
| D | strided_slice_test.cc | 31 std::initializer_list<int> begin_shape, in StridedSliceOpModel() argument
46 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1420 ShapeHandle begin_shape; in SliceShape() local
1421 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &begin_shape)); in SliceShape()
1426 TF_RETURN_IF_ERROR(c->Merge(begin_shape, sizes_shape, &begin_shape)); in SliceShape()
1428 DimensionHandle ndims = c->Dim(begin_shape, 0); in SliceShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1567 ShapeHandle begin_shape, end_shape, strides_shape; in __anon7c94107b2502() local
1568 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &begin_shape)); in __anon7c94107b2502()
1571 TF_RETURN_IF_ERROR(c->Merge(begin_shape, end_shape, &begin_shape)); in __anon7c94107b2502()
1572 TF_RETURN_IF_ERROR(c->Merge(begin_shape, strides_shape, &begin_shape)); in __anon7c94107b2502()
1573 DimensionHandle sparse_dims_dim = c->Dim(begin_shape, 0); in __anon7c94107b2502()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1680 std::initializer_list<int> begin_shape, in StridedSliceOpModel() argument
1689 begin_ = AddConstInput(TensorType_INT32, begin_data, begin_shape); in StridedSliceOpModel()
1698 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|