Searched defs:end_tensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 37 const Tensor* end_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 | 35 const auto& end_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | slice_ops.cc | 202 nvinfer1::ITensor* end_tensor = (*end_const)->getOutput(0); in HandleDynamicStridedSliceInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils.cc | 173 auto end_tensor = CreateI32DenseConst(builder, end, loc); in CreateStridedSliceOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 2589 Tensor end_tensor = tensor::DeepCopy(size_weights.GetTensor()); in ConvertSlice() local
|