Searched defs:strides_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 37 const Tensor& strides_tensor; member
171 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp()
419 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 214 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
392 Tensor strides_tensor; in CompileAsDynamicUpdateSlice() local
495 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
627 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 69 Tensor strides_tensor; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 171 auto strides_tensor = CreateI32DenseConst(builder, strides, loc); in CreateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 1199 Tensor ksizes_tensor, strides_tensor; in TEST() local
|
| D | common_shape_fns.cc | 1645 const Tensor* strides_tensor = c->input_tensor(c->num_inputs() - 1); in MaxPoolV2Shape() local
|