Searched refs:strides_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 37 const Tensor& strides_tensor; member
95 const T* const strides_flat = sparse.strides_tensor.vec<T>().data(); in BuildDenseSpec()
171 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp() argument
181 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
186 end_tensor->NumElements() == strides_tensor.NumElements()); in ValidateStridedSliceOp()
188 !TensorShapeUtils::IsVector(strides_tensor.shape())) { in ValidateStridedSliceOp()
194 strides_tensor.shape().DebugString(), " instead."); in ValidateStridedSliceOp()
198 "but got shape ", strides_tensor.shape().DebugString(), in ValidateStridedSliceOp()
216 StridedSliceSparseSpec sparse_spec = {strides_tensor.NumElements(), in ValidateStridedSliceOp()
220 strides_tensor, in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 66 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
78 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 214 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
224 &strides_tensor)); in Compile()
233 end_is_constant ? &end_tensor : nullptr, strides_tensor, in Compile()
392 Tensor strides_tensor; in CompileAsDynamicUpdateSlice() local
399 &strides_tensor)); in CompileAsDynamicUpdateSlice()
402 nullptr, nullptr, strides_tensor, input_shape, begin_mask_, in CompileAsDynamicUpdateSlice()
495 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
501 &strides_tensor)); in Compile()
506 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
627 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 69 Tensor strides_tensor; in Compute() local
73 &strides_tensor)); in Compute()
75 auto strides = strides_tensor.vec<Tidx>(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 1199 Tensor ksizes_tensor, strides_tensor; in TEST() local
1200 auto set_op = [&op, &ksizes_tensor, &strides_tensor]( in TEST()
1215 strides_tensor = test::AsTensor<int32>(strides); in TEST()
1216 op.input_tensors[2] = &strides_tensor; in TEST()
|
| D | common_shape_fns.cc | 1645 const Tensor* strides_tensor = c->input_tensor(c->num_inputs() - 1); in MaxPoolV2Shape() local
1646 if (strides_tensor == nullptr) { in MaxPoolV2Shape()
1650 strides.resize(strides_tensor->shape().num_elements()); in MaxPoolV2Shape()
1651 auto strides_vec = strides_tensor->flat<int32>(); in MaxPoolV2Shape()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 171 auto strides_tensor = CreateI32DenseConst(builder, strides, loc); in CreateStridedSliceOp() local
174 loc, output_type, input, begin_tensor, end_tensor, strides_tensor, in CreateStridedSliceOp()
|