Searched defs:final_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 94 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
|
| D | strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 77 TensorShape final_shape; in Compile() local
204 TensorShape processing_shape, final_shape; in Compile() local
315 TensorShape final_shape; in Compile() local
|
| D | reduction_ops_common.cc | 87 std::vector<int64> final_shape; in Compile() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 157 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
380 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 173 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1690 PartialTensorShape processing_shape, final_shape; in __anonf6523ebd2402() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2440 const nvinfer1::Dims* final_shape = nullptr) { in ConvertStridedSliceHelper()
2727 PartialTensorShape final_shape; in ConvertStridedSlice() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3221 TensorShape final_shape; in GetSliceAxis() local
|