Home
last modified time | relevance | path

Searched refs:MakeShapeFromShapeTensor (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dstateful_random_ops.cc30 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in StatefulRandomShape()
78 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anon0d38c5e20102()
116 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon0d38c5e20402()
135 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anon0d38c5e20502()
154 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon0d38c5e20602()
Drandom_ops.cc145 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon2645a1db0402()
169 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon2645a1db0502()
188 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon2645a1db0602()
Dstateless_random_ops.cc34 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShape()
157 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anona1237d120302()
Dnccl_ops.cc132 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonf492395a0102()
Dnn_ops.cc375 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anonad10f3360402()
589 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anonad10f3360802()
608 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anonad10f3360902()
690 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anonad10f3360c02()
708 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anonad10f3360d02()
1437 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonad10f3362202()
1825 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anonad10f3362b02()
1852 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anonad10f3362c02()
1891 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in __anonad10f3362d02()
1936 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in __anonad10f3362e02()
[all …]
Dcollective_ops.cc174 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(/*input_idx=*/3, &out)); in __anon52c2f9040302()
Dstateless_random_ops_v2.cc39 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShapeV2()
Dmkl_nn_ops.cc67 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anondbc33bd90102()
93 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anondbc33bd90202()
133 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anondbc33bd90302()
152 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anondbc33bd90402()
1264 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anondbc33bd91202()
1287 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anondbc33bd91302()
Dsparse_ops.cc102 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &a_shape)); in __anon36a8fc880402()
202 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon36a8fc880902()
626 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &constant_input_shape)); in __anon36a8fc881802()
Dsparse_csr_matrix_ops.cc70 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &dense_shape)); in __anon59a9d6c30102()
460 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &dense_shape)); in __anon59a9d6c30b02()
Darray_ops.cc190 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in SetOutputShapeForReshape()
408 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon42d741190402()
478 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon42d741190702()
1102 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon42d741191602()
1777 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon42d741192602()
1844 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &multiples)); in __anon42d741192702()
3145 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output_shape)); in __anon42d741194802()
Dimage_ops.cc920 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(3, &out)); in __anon30b5031d1702()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc227 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
228 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
253 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
254 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
1069 TEST_F(ShapeInferenceTest, MakeShapeFromShapeTensor) { in TEST_F() argument
1075 Status s = c.MakeShapeFromShapeTensor(0, &out); in TEST_F()
1132 c.MakeShapeFromShapeTensor(0, &out).error_message()); in TEST_F()
Dcommon_shape_fns.cc879 c->MakeShapeFromShapeTensor(0, &specified_input_grad_shape)); in Conv2DBackpropInputShape()
1125 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPoolGradShape()
1792 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPool3DGradShape()
2084 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in RandomShape()
2169 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &begin_value)); in SliceShape()
2203 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &sizes_value)); in SliceShape()
Dshape_inference.h465 Status MakeShapeFromShapeTensor(int input_idx, ShapeHandle* out);
Dshape_inference.cc736 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc833 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0 /* input_idx */, &out)); in TensorAsShapeShapeFn()