Searched refs:strides_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 35 const Tensor& strides_tensor; member
86 const T* const strides_flat = sparse.strides_tensor.vec<T>().data(); in BuildDenseSpec()
154 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp() argument
164 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
169 end_tensor->NumElements() == strides_tensor.NumElements()); in ValidateStridedSliceOp()
171 !TensorShapeUtils::IsVector(strides_tensor.shape())) { in ValidateStridedSliceOp()
177 strides_tensor.shape().DebugString(), " instead."); in ValidateStridedSliceOp()
181 "but got shape ", strides_tensor.shape().DebugString(), in ValidateStridedSliceOp()
199 StridedSliceSparseSpec sparse_spec = {strides_tensor.NumElements(), in ValidateStridedSliceOp()
203 strides_tensor, in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 45 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
56 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 58 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
64 &strides_tensor)); in Compile()
70 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
142 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
148 &strides_tensor)); in Compile()
153 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
250 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
256 &strides_tensor)); in Compile()
268 &begin_tensor, &end_tensor, strides_tensor, lhs_shape, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 59 Tensor strides_tensor; in Compute() local
63 &strides_tensor)); in Compute()
65 auto strides = strides_tensor.vec<Tidx>(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 964 Tensor ksizes_tensor, strides_tensor; in TEST() local
965 auto set_op = [&op, &ksizes_tensor, &strides_tensor]( in TEST()
980 strides_tensor = test::AsTensor<int32>(strides); in TEST()
981 op.input_tensors[2] = &strides_tensor; in TEST()
|
| D | common_shape_fns.cc | 983 const Tensor* strides_tensor = c->input_tensor(c->num_inputs() - 1); in MaxPoolV2Shape() local
984 if (strides_tensor == nullptr) { in MaxPoolV2Shape()
988 strides.resize(strides_tensor->shape().num_elements()); in MaxPoolV2Shape()
989 auto strides_vec = strides_tensor->flat<int32>(); in MaxPoolV2Shape()
|