| /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 __anona601b7c40102()
78 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anona601b7c40202()
96 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anona601b7c40302()
|
| D | random_ops.cc | 135 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon979c5d7d0402()
159 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon979c5d7d0502()
178 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon979c5d7d0602()
|
| D | nn_ops.cc | 90 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon3e672dd80202()
292 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon3e672dd80502()
315 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon3e672dd80602()
507 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon3e672dd80a02()
525 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon3e672dd80b02()
587 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon3e672dd80e02()
605 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon3e672dd80f02()
634 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon3e672dd81002()
1343 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon3e672dd82802()
1707 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon3e672dd83102()
[all …]
|
| D | nccl_ops.cc | 132 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon3461a47c0102()
|
| 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 __anona7ffb82a0402()
202 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anona7ffb82a0902()
586 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &constant_input_shape)); in __anona7ffb82a1602()
|
| D | mkl_nn_ops.cc | 733 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon4d19f77b0d02()
755 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon4d19f77b0e02()
|
| 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 __anon7c94107b0402()
472 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon7c94107b0702()
1035 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon7c94107b1602()
1638 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon7c94107b2602()
1689 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &multiples)); in __anon7c94107b2702()
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 __anon6a71d27f1402()
|
| /external/tensorflow/tensorflow/contrib/coder/ops/ |
| D | coder_ops.cc | 96 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon8696facd0102()
|
| /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()
|