Searched refs:MakeDimForScalarInput (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | stateless_random_ops.cc | 110 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anonef3c4ef00202()
|
| D | random_ops.cc | 129 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anonb4eee6390302()
|
| D | image_ops.cc | 230 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(3, &output_dim)); in CombinedNMSShapeFn()
235 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &size_per_class)); in CombinedNMSShapeFn()
1027 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anonf6f833bb1b02()
1054 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anonf6f833bb1c02()
|
| D | parsing_ops.cc | 124 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &fixed_length)); in __anon5cc8b68c0302()
|
| D | nn_ops.cc | 1334 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &k_dim)); in TopKShapeFn()
1492 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &n_dim)); in __anon1bbab8942102()
|
| D | array_ops.cc | 2821 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &depth)); in __anon091a71b73b02()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 1292 TEST_F(ShapeInferenceTest, MakeDimForScalarInput) { in TEST_F() argument
1300 TF_EXPECT_OK(c.MakeDimForScalarInput(0, &d)); in TEST_F()
1304 c.MakeDimForScalarInput(1, &d), in TEST_F()
1312 TF_EXPECT_OK(c.MakeDimForScalarInput(0, &d)); in TEST_F()
1316 c.MakeDimForScalarInput(1, &d), in TEST_F()
|
| D | shape_inference.h | 590 Status MakeDimForScalarInput(int idx, DimensionHandle* out);
|
| D | shape_inference.cc | 973 Status InferenceContext::MakeDimForScalarInput(int idx, DimensionHandle* out) { in MakeDimForScalarInput() function in tensorflow::shape_inference::InferenceContext
|
| D | common_shape_fns.cc | 2196 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &num_segments_dim)); in UnsortedSegmentReductionShapeFn()
|