| /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/util/ |
| D | strided_slice_op.cc | 36 const Tensor* begin_tensor; member
97 dense->begin_valid = sparse.begin_tensor != nullptr; in BuildDenseSpec()
100 const T* const begin_flat = sparse.begin_tensor != nullptr in BuildDenseSpec()
101 ? sparse.begin_tensor->vec<T>().data() in BuildDenseSpec()
171 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp() argument
181 begin_tensor != nullptr && in ValidateStridedSliceOp()
182 !(TensorShapeUtils::IsVector(begin_tensor->shape()) && in ValidateStridedSliceOp()
183 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
184 begin_tensor->NumElements() < 32 /* using 32 bit masks */); in ValidateStridedSliceOp()
191 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/compiler/tf2tensorrt/convert/ops/ |
| D | slice_ops.cc | 195 nvinfer1::ITensor* begin_tensor = (*begin_const)->getOutput(0); in HandleDynamicStridedSliceInput() local
223 (*dynamic_begin_masked_tensor)->getOutput(0), begin_tensor); in HandleDynamicStridedSliceInput()
225 begin_tensor = (*begin_corrected)->getOutput(0); in HandleDynamicStridedSliceInput()
232 builder->Sub(end_tensor, begin_tensor); in HandleDynamicStridedSliceInput()
240 slice_layer->setInput(1, *begin_tensor); in HandleDynamicStridedSliceInput()
|
| /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 | 251 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
254 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
268 begin_is_constant ? &begin_tensor : nullptr, in Compile()
534 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
536 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
545 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
666 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
668 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
689 &begin_tensor, &end_tensor, strides_tensor, lhs_shape, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 73 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs() local
92 TensorShapeUtils::IsVector(begin_tensor.shape()) && in ValidateMklInputs()
94 begin_tensor.NumElements() == input_dims && in ValidateMklInputs()
98 input_dims, ", but got shapes ", begin_tensor.shape().DebugString(), in ValidateMklInputs()
101 *begin = IntTensorToInt64Vec(begin_tensor); in ValidateMklInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 172 auto begin_tensor = CreateI32DenseConst(builder, begin, loc); in CreateStridedSliceOp() local
177 loc, output_type, input, begin_tensor, end_tensor, strides_tensor, in CreateStridedSliceOp()
|