Searched defs:final_shape (Results 1 – 10 of 10) 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 | 106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 58 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
617 TensorShape final_shape; in Compile() local
|
| D | reduction_ops_common.cc | 92 std::vector<int64> final_shape; in Compile() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 174 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
422 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp()
|
| /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 | 303 SmallVector<int64_t, 4> final_shape = { in HandleConv2DFilter() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1740 PartialTensorShape processing_shape, final_shape; in __anondb9326b22502() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2849 const nvinfer1::Dims* final_shape = nullptr, in ConvertStridedSliceHelper()
3142 PartialTensorShape final_shape; in ConvertStridedSlice() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3367 TensorShape final_shape; in GetStridedSliceAxis() local
|