Searched defs:strides_tensor (Results 1 – 7 of 7) sorted by relevance
| /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
429 Tensor strides_tensor; in CompileAsDynamicUpdateSlice() 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/core/util/ |
| H A D | strided_slice_op.cc | 38 const Tensor& strides_tensor; member
172 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp()
424 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | unravel_index_op.cc | 101 Tensor strides_tensor; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils.cc | 174 auto strides_tensor = CreateI32DenseConst(builder, strides, loc); in CreateStridedSliceOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns_test.cc | 1205 Tensor ksizes_tensor, strides_tensor; in TEST() local
|
| H A D | common_shape_fns.cc | 1736 const Tensor* strides_tensor = c->input_tensor(c->num_inputs() - 1); in MaxPoolV2Shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 2588 Tensor strides_tensor = tensor::DeepCopy(begin_weights.GetTensor()); in ConvertSlice() local
|