| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | canonicalize_all_gather_for_cse.cc | 36 const Shape& out_shape = inst->shape(); in IsAddingOnlyDegenerateDimensions() local
59 const Shape& out_shape = ag->shape(); in RunOnComputation() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | batch_ops.cc | 103 shape_inference::ShapeHandle out_shape; in __anon720d5c150202() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | diag_op.cc | 53 TensorShape out_shape; in Compute() local
93 TensorShape out_shape; in Compute() local
|
| D | decode_padded_raw_op.cc | 63 TensorShape out_shape = input.shape(); in Compute() local
|
| D | maxpooling_op.cc | 811 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
894 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
962 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1082 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute() local
1140 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1200 TensorShape out_shape = in Compute() local
1328 TensorShape out_shape = in Compute() local
|
| D | decode_raw_op.cc | 57 TensorShape out_shape = input.shape(); in Compute() local
|
| D | nth_element_op.cc | 67 TensorShape out_shape; in Compute() local
|
| D | reduction_ops_common.cc | 29 TensorShape ReductionHelper::out_shape() const { in out_shape() function in tensorflow::ReductionHelper
|
| D | matmul_op_fused.cc | 197 TensorShape out_shape( in Compute() local
|
| D | quantized_matmul_op.cc | 117 TensorShape out_shape( in Compute() local
|
| D | cudnn_pooling_gpu.cc | 42 const auto out_shape = output->shape(); in Compute() local
|
| D | sparse_tensor_dense_matmul_op.cc | 117 TensorShape out_shape({outer_left, outer_right}); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 104 TensorShape out_shape = bcast.output_batch_shape(); in Compute() local
179 const TensorShape& out_shape) { in CreateMatMulParams()
|
| D | mkl_matmul_op.cc | 69 TensorShape out_shape( in Compute() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_argmax_cuda.cu | 79 array<DenseIndex, 3> out_shape; in test_cuda_argmax_dim() local
168 array<DenseIndex, 3> out_shape; in test_cuda_argmin_dim() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | merge_reshape_into_preceding_transpose.cc | 67 std::vector<int> out_shape = output_array.shape().dims(); in ReshapeToTranspose() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 345 const auto& out_shape = out->shape(); in ValidateAndInitializeToDense() local
379 const auto& out_shape = out->shape(); in ToDense() local
395 const auto& out_shape = out->shape().dim_sizes(); in ToDense() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 149 TensorShape out_shape; in ComputeAsync() local
289 TensorShape out_shape; in ComputeAsync() local
|
| D | banded_triangular_solve_op.cc | 228 TensorShape out_shape = bcast.output_batch_shape(); in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test_base.py | 135 out_shape, argument
237 def _gpuVsCpuCase(self, in_shape, out_shape, align_corners, argument
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.cc | 100 TensorShape out_shape(out_sizes); in Compute() local
|
| D | extract_volume_patches_op.cc | 127 TensorShape out_shape(out_sizes); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 97 TensorShape out_shape({batch, out_rows, out_cols, out_depth}); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 80 llvm::SmallVector<int64_t, 4> out_shape(shape_x.size()); in GetBroadcastType() local
112 llvm::SmallVector<int64_t, 4> out_shape(shape_large.begin(), in GetBroadcastType() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 79 TensorShape out_shape; in XlaGather() local
|