Home
last modified time | relevance | path

Searched refs:MakeShapeFromShapeTensor (Results 1 – 18 of 18) 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 __anon57f0bee70102()
116 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon57f0bee70402()
135 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anon57f0bee70502()
154 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon57f0bee70602()
Drandom_ops.cc145 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon81b079400402()
169 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon81b079400502()
188 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon81b079400602()
Dstateless_random_ops.cc34 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShape()
157 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon42da96b70302()
Dnccl_ops.cc132 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon1c25f37f0102()
Dnn_ops.cc375 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon97b5e01b0402()
589 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon97b5e01b0802()
608 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon97b5e01b0902()
690 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon97b5e01b0c02()
708 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon97b5e01b0d02()
1437 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon97b5e01b2202()
1807 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon97b5e01b2902()
1834 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon97b5e01b2a02()
1873 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in __anon97b5e01b2b02()
1950 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon97b5e01b2c02()
[all …]
Dmkl_nn_ops.cc67 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon372e133e0102()
93 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon372e133e0202()
133 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon372e133e0302()
152 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon372e133e0402()
1098 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon372e133e1002()
1129 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon372e133e1102()
Dstateless_random_ops_v2.cc39 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShapeV2()
Dcollective_ops.cc183 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(/*input_idx=*/3, &out)); in __anon8ce4fae90302()
Dsparse_ops.cc103 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &a_shape)); in __anon9213d3ed0402()
203 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon9213d3ed0902()
629 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &constant_input_shape)); in __anon9213d3ed1802()
Dsparse_csr_matrix_ops.cc70 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &dense_shape)); in __anon324425080102()
460 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &dense_shape)); in __anon324425080b02()
Darray_ops.cc190 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in SetOutputShapeForReshape()
408 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon5ce23fde0402()
478 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon5ce23fde0702()
1112 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon5ce23fde1602()
1790 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon5ce23fde2602()
1857 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &multiples)); in __anon5ce23fde2702()
3141 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output_shape)); in __anon5ce23fde4702()
Dimage_ops.cc920 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(3, &out)); in __anon4ac001e21702()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc228 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
229 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
254 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
255 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
1084 TEST_F(ShapeInferenceTest, MakeShapeFromShapeTensor) { in TEST_F() argument
1090 Status s = c.MakeShapeFromShapeTensor(0, &out); in TEST_F()
1147 c.MakeShapeFromShapeTensor(0, &out).error_message()); in TEST_F()
Dcommon_shape_fns.cc900 c->MakeShapeFromShapeTensor(0, &specified_input_grad_shape)); in Conv2DBackpropInputShape()
955 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in Conv2DBackpropFilterWithBiasShape()
1165 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPoolGradShape()
1832 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPool3DGradShape()
2124 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in RandomShape()
2209 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &begin_value)); in SliceShape()
2243 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &sizes_value)); in SliceShape()
Dshape_inference.h475 Status MakeShapeFromShapeTensor(int input_idx, ShapeHandle* out);
Dshape_inference.cc743 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc391 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &size_indices_value)); in __anon592c2f360202()
826 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output)); in __anon592c2f360b02()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc833 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0 /* input_idx */, &out)); in TensorAsShapeShapeFn()