Home
last modified time | relevance | path

Searched defs:processing_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc94 TensorShape processing_shape, final_shape; in Compute() local
202 TensorShape processing_shape, final_shape; in Compute() local
296 TensorShape processing_shape, final_shape; in Compute() local
Dstrided_slice_op_impl.h83 const TensorShape& processing_shape, in HandleStridedSliceCase()
121 const TensorShape& processing_shape, in HandleStridedSliceGradCase()
Dstrided_slice_op_test.cc106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc58 TensorShape processing_shape, TensorShape final_shape, in EmitDynamicSlice()
225 TensorShape processing_shape, final_shape; in Compile() local
393 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
479 TensorShape processing_shape, final_shape; in Compile() local
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc174 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
421 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1740 PartialTensorShape processing_shape, final_shape; in __anondb9326b22502() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3366 TensorShape processing_shape; in GetStridedSliceAxis() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3141 PartialTensorShape processing_shape; in ConvertStridedSlice() local