Searched refs:strides_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 38 const Tensor& strides_tensor; member
96 const T* const strides_flat = sparse.strides_tensor.vec<T>().data(); in BuildDenseSpec()
172 const Tensor& strides_tensor, const PartialTensorShape& input_shape, in ValidateStridedSliceOp() argument
183 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
188 end_tensor->NumElements() == strides_tensor.NumElements()); in ValidateStridedSliceOp()
190 !TensorShapeUtils::IsVector(strides_tensor.shape())) { in ValidateStridedSliceOp()
196 strides_tensor.shape().DebugString(), " instead."); in ValidateStridedSliceOp()
200 "but got shape ", strides_tensor.shape().DebugString(), in ValidateStridedSliceOp()
218 StridedSliceSparseSpec sparse_spec = {strides_tensor.NumElements(), in ValidateStridedSliceOp()
222 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 | 251 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
261 &strides_tensor)); in Compile()
269 end_is_constant ? &end_tensor : nullptr, strides_tensor, in Compile()
429 Tensor strides_tensor; in CompileAsDynamicUpdateSlice() local
436 &strides_tensor)); in CompileAsDynamicUpdateSlice()
439 nullptr, nullptr, strides_tensor, input_shape, begin_mask_, in CompileAsDynamicUpdateSlice()
534 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
540 &strides_tensor)); in Compile()
545 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
666 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 101 Tensor strides_tensor; in Compute() local
105 &strides_tensor)); in Compute()
107 auto strides = strides_tensor.vec<Tidx>(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 1205 Tensor ksizes_tensor, strides_tensor; in TEST() local
1206 auto set_op = [&op, &ksizes_tensor, &strides_tensor]( in TEST()
1221 strides_tensor = test::AsTensor<int32>(strides); in TEST()
1222 op.input_tensors[2] = &strides_tensor; in TEST()
|
| D | common_shape_fns.cc | 1736 const Tensor* strides_tensor = c->input_tensor(c->num_inputs() - 1); in MaxPoolV2Shape() local
1737 if (strides_tensor == nullptr) { in MaxPoolV2Shape()
1741 strides.resize(strides_tensor->shape().num_elements()); in MaxPoolV2Shape()
1742 auto strides_vec = strides_tensor->flat<int32>(); in MaxPoolV2Shape()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 174 auto strides_tensor = CreateI32DenseConst(builder, strides, loc); in CreateStridedSliceOp() local
177 loc, output_type, input, begin_tensor, end_tensor, strides_tensor, in CreateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2588 Tensor strides_tensor = tensor::DeepCopy(begin_weights.GetTensor()); in ConvertSlice() local
2594 auto strides_vec = strides_tensor.flat<int32>(); in ConvertSlice()
2621 &begin_weights.GetTensor(), &end_tensor, strides_tensor, input_shape, in ConvertSlice()
|