| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 59 void TestStridedSliceFloat(const int* input_shape, const int* begin_shape, in TestStridedSliceFloat() argument
68 TfLiteIntArray* begin_dims = IntArrayFromInts(begin_shape); in TestStridedSliceFloat()
91 const int* input_shape, const int* begin_shape, const int* end_shape, in TestStridedSliceQuantized() argument
97 TfLiteIntArray* begin_dims = IntArrayFromInts(begin_shape); in TestStridedSliceQuantized()
128 const int begin_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()
149 const int begin_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()
170 const int begin_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/lite/kernels/ |
| D | slice_test.cc | 40 std::initializer_list<int> begin_shape, in SliceOpModel() argument
52 AddConstInput(GetTensorType<index_type>(), begin_data, begin_shape); in SliceOpModel()
58 BuildInterpreter({input_shape, begin_shape, size_shape}); in SliceOpModel()
|
| D | strided_slice_test.cc | 34 std::initializer_list<int> begin_shape, in StridedSliceOpModel() argument
49 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 638 const ArrayRef<int64_t> begin_shape = begin_type.getShape(); in RewriteEllipsisMask() local
639 const int begin_dim = begin_shape.size(); in RewriteEllipsisMask()
643 const int ellipsis_filled_dim_size = input_size - begin_shape[0] + 1; in RewriteEllipsisMask()
685 for (; index < begin_shape[0];) { in RewriteEllipsisMask()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 129 ArrayRef<int64_t> begin_shape, in SliceRankedTensor() argument
147 CreateI64DenseConst(builder, begin_shape, begin_values, location); in SliceRankedTensor()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 199 const TensorShape begin_shape = ctx->InputShape("begin"); in Compile() local
201 ctx, begin_shape.dims() == 1, in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1706 ShapeHandle begin_shape, end_shape, strides_shape; in __anon42d741192502() local
1707 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &begin_shape)); in __anon42d741192502()
1710 TF_RETURN_IF_ERROR(c->Merge(begin_shape, end_shape, &begin_shape)); in __anon42d741192502()
1711 TF_RETURN_IF_ERROR(c->Merge(begin_shape, strides_shape, &begin_shape)); in __anon42d741192502()
1712 DimensionHandle sparse_dims_dim = c->Dim(begin_shape, 0); in __anon42d741192502()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2153 ShapeHandle begin_shape; in SliceShape() local
2154 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &begin_shape)); in SliceShape()
2159 TF_RETURN_IF_ERROR(c->Merge(begin_shape, sizes_shape, &begin_shape)); in SliceShape()
2161 DimensionHandle ndims = c->Dim(begin_shape, 0); in SliceShape()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2511 std::initializer_list<int> begin_shape, in StridedSliceOpModel() argument
2520 begin_ = AddConstInput(TensorType_INT32, begin_data, begin_shape); in StridedSliceOpModel()
2530 {input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 972 TensorShape begin_shape({shape.dims()}); in CreateSliceNode() local
973 begin_shape.AsProto(begin_tensor_proto.mutable_tensor_shape()); in CreateSliceNode()
|