Searched defs:begin_tensor (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 36 const Tensor* begin_tensor; member
171 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp()
423 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 251 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
534 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
666 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | strided_slice_builder.cc | 34 const auto& begin_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
|
| H A D | slice_builder.cc | 44 const auto& begin_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | slice_ops.cc | 195 nvinfer1::ITensor* begin_tensor = (*begin_const)->getOutput(0); in HandleDynamicStridedSliceInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | slice_op.cc | 68 const Tensor& begin_tensor = context->input(1); in SharedSliceValidation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_slice_op.cc | 73 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils.cc | 172 auto begin_tensor = CreateI32DenseConst(builder, begin, loc); in CreateStridedSliceOp() local
|