Home
last modified time | relevance | path

Searched refs:processing_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc94 TensorShape processing_shape, final_shape; in Compute() local
107 &processing_shape, &final_shape, &is_identity, in Compute()
140 const int processing_dims = processing_shape.dims(); in Compute()
142 if (processing_shape.num_elements() > 0) { in Compute()
150 input_dims == 2 && processing_shape.dims() == 2 && in Compute()
161 processing_shape, is_simple_slice, \ in Compute()
202 TensorShape processing_shape, final_shape; in Compute() local
233 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
246 const int processing_dims = processing_shape.dims(); in Compute()
250 if (processing_shape.dims() == 0) { in Compute()
[all …]
Dstrided_slice_op_impl.h45 const TensorShape& processing_shape,
53 const TensorShape& processing_shape,
62 const TensorShape& processing_shape, bool is_simple_slice,
83 const TensorShape& processing_shape, in HandleStridedSliceCase() argument
87 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceCase()
121 const TensorShape& processing_shape, in HandleStridedSliceGradCase() argument
123 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceGradCase()
145 const TensorShape& processing_shape, bool is_simple_slice, Tensor* result) { in operator()
146 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in operator()
169 const TensorShape& processing_shape, bool is_simple_slice, in operator()
[all …]
Dstrided_slice_op_test.cc106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
117 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc58 TensorShape processing_shape, TensorShape final_shape, in EmitDynamicSlice() argument
97 ctx, partial_processing_shape.AsTensorShape(&processing_shape), in EmitDynamicSlice()
186 xla::DynamicSlice(slice, start_indices, processing_shape.dim_sizes()); in EmitDynamicSlice()
225 TensorShape processing_shape, final_shape; in Compile() local
350 EmitDynamicSlice(ctx, strides, processing_shape, final_shape, in Compile()
393 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
404 &processing_shape, &final_shape, &dummy, &dummy, &dummy, in CompileAsDynamicUpdateSlice()
406 for (int64 i = 0; i < processing_shape.dims(); ++i) { in CompileAsDynamicUpdateSlice()
424 for (int64 i = 0; i < processing_shape.dims(); ++i) { in CompileAsDynamicUpdateSlice()
425 if (processing_shape.dim_size(i) == -1) { in CompileAsDynamicUpdateSlice()
[all …]
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc174 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp() argument
272 processing_shape->Clear(); in ValidateStridedSliceOp()
283 processing_shape->AddDim(shrink_i ? 1 : -1); in ValidateStridedSliceOp()
370 processing_shape->AddDim(size_i); in ValidateStridedSliceOp()
372 processing_shape->AddDim(-1); in ValidateStridedSliceOp()
400 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()
421 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
435 if (!partial_processing_shape.AsTensorShape(processing_shape) || in ValidateStridedSliceOp()
Dstrided_slice_op.h69 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
80 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1740 PartialTensorShape processing_shape, final_shape; in __anon42d741192502() local
1746 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anon42d741192502()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3366 TensorShape processing_shape; in GetStridedSliceAxis() local
3377 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetStridedSliceAxis()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3141 PartialTensorShape processing_shape; in ConvertStridedSlice() local
3152 ellipsis_mask, new_axis_mask, shrink_axis_mask, &processing_shape, in ConvertStridedSlice()