/external/tensorflow/tensorflow/core/kernels/ |
D | strided_slice_op.cc | 94 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 …]
|
D | strided_slice_op_impl.h | 45 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 …]
|
D | strided_slice_op_test.cc | 106 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/ |
D | strided_slice_op.cc | 58 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/ |
D | strided_slice_op.cc | 174 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()
|
D | strided_slice_op.h | 69 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, 80 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_ops.cc | 1740 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/ |
D | arithmetic_optimizer.cc | 3366 TensorShape processing_shape; in GetStridedSliceAxis() local 3377 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetStridedSliceAxis()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 3141 PartialTensorShape processing_shape; in ConvertStridedSlice() local 3152 ellipsis_mask, new_axis_mask, shrink_axis_mask, &processing_shape, in ConvertStridedSlice()
|