Searched defs:final_shape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 97 TensorShape processing_shape, final_shape; in Compute() local
205 TensorShape processing_shape, final_shape; in Compute() local
300 TensorShape final_shape; // After reshaping input. in Compute() local
|
| D | strided_slice_op_test.cc | 106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 84 TensorShape final_shape; in EmitDynamicSlice() local
282 TensorShape final_shape; in Compile() local
430 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
516 TensorShape processing_shape, final_shape; in Compile() local
656 TensorShape final_shape; in Compile() local
|
| D | reduction_ops_common.cc | 94 std::vector<int64_t> final_shape; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | slice_ops.cc | 55 std::optional<nvinfer1::Dims> final_shape, std::optional<int> op_instance, in ConvertStridedSliceHelper()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 175 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
427 TensorShape* processing_shape, TensorShape* final_shape, bool* is_identity, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | collectives.cc | 576 llvm::SmallVector<int64_t, 4> final_shape(input_tensor_shape.begin(), in EmitHaloExchange() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 178 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 258 SmallVector<int64_t, 4> final_shape = { in HandleConv2DFilter() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1773 PartialTensorShape processing_shape, final_shape; in __anon091a71b72502() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2575 PartialTensorShape final_shape; in ConvertSlice() local
2702 PartialTensorShape final_shape; in ConvertStridedSlice() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3951 TensorShape final_shape; in GetStridedSliceAxis() local
|