| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 35 const Tensor* begin_tensor; member
96 dense->begin_valid = sparse.begin_tensor != nullptr; in BuildDenseSpec()
99 const T* const begin_flat = sparse.begin_tensor != nullptr in BuildDenseSpec()
100 ? sparse.begin_tensor->vec<T>().data() in BuildDenseSpec()
170 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp() argument
179 begin_tensor != nullptr && in ValidateStridedSliceOp()
180 !(TensorShapeUtils::IsVector(begin_tensor->shape()) && in ValidateStridedSliceOp()
181 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
182 begin_tensor->NumElements() < 32 /* using 32 bit masks */); in ValidateStridedSliceOp()
189 if (begin_tensor != nullptr && end_tensor != nullptr) { in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 65 const Tensor* begin_tensor, const Tensor* end_tensor,
77 const Tensor* begin_tensor, const Tensor* end_tensor,
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | slice_builder.cc | 44 const auto& begin_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
47 if (begin_tensor.type == kTfLiteInt32) { in PopulateSubGraph()
48 GetBeginAndSizeVectors<int>(input_tensor.dims->size, &begin_tensor, in PopulateSubGraph()
50 } else if (begin_tensor.type == kTfLiteInt64) { in PopulateSubGraph()
51 GetBeginAndSizeVectors<int64_t>(input_tensor.dims->size, &begin_tensor, in PopulateSubGraph()
|
| D | strided_slice_builder.cc | 34 const auto& begin_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
38 graph_builder_->AddConstNodeWithData(inputs->data[1], begin_tensor); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 68 const Tensor& begin_tensor = context->input(1); in SharedSliceValidation() local
73 TensorShapeUtils::IsVector(begin_tensor.shape()) && in SharedSliceValidation()
75 begin_tensor.NumElements() == input.dims() && in SharedSliceValidation()
79 input.dims(), ", but got shapes ", begin_tensor.shape().DebugString(), in SharedSliceValidation()
83 IntTensorToInt64Vec(begin_tensor, begin); in SharedSliceValidation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 214 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
217 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
232 begin_is_constant ? &begin_tensor : nullptr, in Compile()
495 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
497 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
506 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
627 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
629 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
650 &begin_tensor, &end_tensor, strides_tensor, lhs_shape, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 70 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs() local
89 TensorShapeUtils::IsVector(begin_tensor.shape()) && in ValidateMklInputs()
91 begin_tensor.NumElements() == input_dims && in ValidateMklInputs()
95 input_dims, ", but got shapes ", begin_tensor.shape().DebugString(), in ValidateMklInputs()
98 *begin = IntTensorToInt64Vec(begin_tensor); in ValidateMklInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 169 auto begin_tensor = CreateI32DenseConst(builder, begin, loc); in CreateStridedSliceOp() local
174 loc, output_type, input, begin_tensor, end_tensor, strides_tensor, in CreateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 972 Tensor begin_tensor; in TEST_F() local
973 EXPECT_TRUE(begin_tensor.FromProto( in TEST_F()
977 test::ExpectTensorEqual<int>(begin_tensor_expected, begin_tensor); in TEST_F()
|