Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dstateless_random_ops.cc110 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anona1237d120202()
Drandom_ops.cc129 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &num_samples)); in __anon2645a1db0302()
Dimage_ops.cc230 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 __anon30b5031d1b02()
1054 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anon30b5031d1c02()
Dparsing_ops.cc122 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &fixed_length)); in __anonf8f6e66e0302()
Dnn_ops.cc1300 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &k_dim)); in TopKShapeFn()
1363 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &n_dim)); in __anonad10f3362002()
Darray_ops.cc2788 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &depth)); in __anon42d741193b02()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc1253 TEST_F(ShapeInferenceTest, MakeDimForScalarInput) { in TEST_F() argument
1261 EXPECT_TRUE(c.MakeDimForScalarInput(0, &d).ok()); in TEST_F()
1264 EXPECT_CONTAINS(c.MakeDimForScalarInput(1, &d).error_message(), in TEST_F()
1271 EXPECT_TRUE(c.MakeDimForScalarInput(0, &d).ok()); in TEST_F()
1274 EXPECT_CONTAINS(c.MakeDimForScalarInput(1, &d).error_message(), in TEST_F()
Dshape_inference.h504 Status MakeDimForScalarInput(int idx, DimensionHandle* out);
Dshape_inference.cc940 Status InferenceContext::MakeDimForScalarInput(int idx, DimensionHandle* out) { in MakeDimForScalarInput() function in tensorflow::shape_inference::InferenceContext
Dcommon_shape_fns.cc2105 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &num_segments_dim)); in UnsortedSegmentReductionShapeFn()