Home
last modified time | relevance | path

Searched defs:out_shape (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dbatch_ops.cc64 shape_inference::ShapeHandle out_shape; in __anona3c1e4520202() local
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc53 TensorShape out_shape; in Compute() local
93 TensorShape out_shape; in Compute() local
Ddecode_raw_op.cc52 TensorShape out_shape = input.shape(); in Compute() local
Dreduction_ops_common.cc29 TensorShape ReductionHelper::out_shape() const { in out_shape() function in tensorflow::ReductionHelper
Dnth_element_op.cc67 TensorShape out_shape; in Compute() local
Dmaxpooling_op.cc775 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
853 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
903 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
961 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute() local
1011 TensorShape out_shape({params.tensor_in_batch, params.out_height, in Compute() local
1068 TensorShape out_shape = in Compute() local
1180 TensorShape out_shape = in Compute() local
Ddeterminant_op.cc149 TensorShape out_shape; in ComputeAsync() local
289 TensorShape out_shape; in ComputeAsync() local
Dextract_image_patches_op.cc97 TensorShape out_shape(out_sizes); in Compute() local
Dquantized_matmul_op.cc117 TensorShape out_shape( in Compute() local
Dcudnn_pooling_gpu.cc41 const auto out_shape = output->shape(); in Compute() local
Dexample_parsing_ops.cc449 TensorShape out_shape; in Compute() local
564 TensorShape out_shape; in Compute() local
Dmkl_matmul_op.cc66 TensorShape out_shape( in Compute() local
Dsparse_tensor_dense_matmul_op.cc116 TensorShape out_shape({outer_left, outer_right}); in Compute() local
Ddepthwise_conv_op.cc348 TensorShape out_shape = in Compute() local
Dconv_ops_3d.cc125 TensorShape out_shape = ShapeFromFormat( in Compute() local
Dmkl_batch_matmul_op.cc68 TensorShape out_shape; in Compute() local
Dmkl_conv_ops.h252 TensorShape out_shape = in GetOutputAndPadSizeInMklOrder() local
Dmatmul_op.cc476 TensorShape out_shape( in Compute() local
/external/eigen/unsupported/test/
Dcxx11_tensor_argmax_cuda.cu79 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/core/kernels/neon/
Dneon_depthwise_conv_op.cc97 TensorShape out_shape({batch, out_rows, out_cols, out_depth}); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc71 TensorShape out_shape; in XlaGather() local
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h375 const auto& out_shape = out->shape(); in ValidateAndInitializeToDense() local
398 const auto& out_shape = out->shape(); in ToDense() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcopy_test.cc246 Shape out_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {1, 0}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Ddata_feeder.py65 def out_el_shape(out_shape, num_classes): argument
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc345 TensorShape out_shape; in BatchExampleProtoToTensors() local

12