Home
last modified time | relevance | path

Searched refs:strides_tensor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc35 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 …]
Dstrided_slice_op.h45 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
56 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc58 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/
Dunravel_index_op.cc59 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/
Dcommon_shape_fns_test.cc964 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()
Dcommon_shape_fns.cc983 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()