Home
last modified time | relevance | path

Searched refs:MakeDimForScalarInput (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dstateless_random_ops.cc91 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anon62f9110b0202()
Drandom_ops.cc119 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anond67e3a940302()
Dimage_ops.cc170 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(3, &output_dim)); in CombinedNMSShapeFn()
175 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &size_per_class)); in CombinedNMSShapeFn()
856 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anonc970e8b61802()
Dnn_ops.cc1206 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &k_dim)); in TopKShapeFn()
1269 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &n_dim)); in __anon4650176f2602()
Dmath_ops.cc1118 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &num_segments_dim)); in UnsortedSegmentReductionShapeFn()
Darray_ops.cc2692 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &depth)); in __anondb9326b23c02()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc1245 TEST_F(ShapeInferenceTest, MakeDimForScalarInput) { in TEST_F() argument
1253 EXPECT_TRUE(c.MakeDimForScalarInput(0, &d).ok()); in TEST_F()
1257 str_util::StrContains(c.MakeDimForScalarInput(1, &d).error_message(), in TEST_F()
1264 EXPECT_TRUE(c.MakeDimForScalarInput(0, &d).ok()); in TEST_F()
1268 str_util::StrContains(c.MakeDimForScalarInput(1, &d).error_message(), in TEST_F()
Dshape_inference.h514 Status MakeDimForScalarInput(int idx, DimensionHandle* out);
Dshape_inference.cc967 Status InferenceContext::MakeDimForScalarInput(int idx, DimensionHandle* out) { in MakeDimForScalarInput() function in tensorflow::shape_inference::InferenceContext