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()
|