| /external/tensorflow/tensorflow/core/ops/ |
| D | batch_ops.cc | 84 shape_inference::ShapeHandle out_shape; in __anon130e45de0202() 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_raw_op.cc | 52 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 | maxpooling_op.cc | 789 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
867 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
930 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1044 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute() local
1097 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1155 TensorShape out_shape = in Compute() local
1273 TensorShape out_shape = in Compute() local
|
| D | determinant_op.cc | 149 TensorShape out_shape; in ComputeAsync() local
289 TensorShape out_shape; in ComputeAsync() local
|
| D | extract_image_patches_op.cc | 97 TensorShape out_shape(out_sizes); in Compute() local
|
| D | extract_volume_patches_op.cc | 124 TensorShape out_shape(out_sizes); 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 | 116 TensorShape out_shape({outer_left, outer_right}); in Compute() local
|
| D | mkl_matmul_op.cc | 72 TensorShape out_shape( in Compute() local
|
| D | example_parsing_ops.cc | 623 TensorShape out_shape; in Compute() local
738 TensorShape out_shape; in Compute() local
|
| D | conv_ops_3d.cc | 160 TensorShape out_shape = ShapeFromFormat( in Compute() local
|
| D | mkl_batch_matmul_op.cc | 68 TensorShape out_shape; in Compute() local
|
| D | depthwise_conv_op.cc | 355 TensorShape out_shape = in Compute() local
|
| D | mkl_conv_ops.h | 445 TensorShape out_shape = variable
|
| /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/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/tf2xla/kernels/ |
| D | gather_op.cc | 73 TensorShape out_shape; in XlaGather() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 452 const auto& out_shape = out->shape(); in ValidateAndInitializeToDense() local
475 const auto& out_shape = out->shape(); in ToDense() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | copy_test.cc | 246 Shape out_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {1, 0}); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | data_feeder.py | 71 def out_el_shape(out_shape, num_classes): argument
|