| /external/tensorflow/tensorflow/core/ops/ |
| D | stateful_random_ops.cc | 28 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in StatefulRandomShape()
64 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anonae3f3ebb0102()
78 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonae3f3ebb0202()
96 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anonae3f3ebb0302()
|
| D | random_ops.cc | 135 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond67e3a940402()
159 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond67e3a940502()
178 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond67e3a940602()
|
| D | nn_ops.cc | 90 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4650176f0202()
292 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4650176f0502()
315 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon4650176f0602()
507 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4650176f0a02()
525 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon4650176f0b02()
587 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4650176f0e02()
605 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon4650176f0f02()
634 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4650176f1002()
1343 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon4650176f2802()
1707 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon4650176f3102()
[all …]
|
| D | nccl_ops.cc | 132 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonda2b4fd30102()
|
| D | stateless_random_ops.cc | 34 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShape()
|
| D | sparse_ops.cc | 102 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &a_shape)); in __anone6e195410402()
202 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anone6e195410902()
586 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &constant_input_shape)); in __anone6e195411602()
|
| D | mkl_nn_ops.cc | 733 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon8bfbd4920d02()
755 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon8bfbd4920e02()
|
| D | array_ops.cc | 183 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in SetOutputShapeForReshape()
402 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anondb9326b20402()
472 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anondb9326b20702()
1035 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anondb9326b21602()
1638 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anondb9326b22602()
1689 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &multiples)); in __anondb9326b22702()
2937 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output_shape)); in ScatterNdShape()
|
| D | image_ops.cc | 752 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(3, &out)); in __anonc970e8b61402()
|
| /external/tensorflow/tensorflow/contrib/coder/ops/ |
| D | coder_ops.cc | 96 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anonfb33fee40102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 216 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
217 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
239 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
240 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
1054 TEST_F(ShapeInferenceTest, MakeShapeFromShapeTensor) { in TEST_F() argument
1060 Status s = c.MakeShapeFromShapeTensor(0, &out); in TEST_F()
1122 c.MakeShapeFromShapeTensor(0, &out).error_message()); in TEST_F()
|
| D | common_shape_fns.cc | 1382 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in RandomShape()
1436 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &begin_value)); in SliceShape()
1470 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &sizes_value)); in SliceShape()
|
| D | shape_inference.h | 479 Status MakeShapeFromShapeTensor(int input_idx, ShapeHandle* out);
|
| D | shape_inference.cc | 793 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 848 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0 /* input_idx */, &out)); in TensorAsShapeShapeFn()
|