Home
last modified time | relevance | path

Searched refs:MakeShapeFromPartialTensorShape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dparsing_ops.cc89 c->MakeShapeFromPartialTensorShape(attrs.dense_shapes[i], &dense)); in __anon952516500202()
132 c->MakeShapeFromPartialTensorShape(attrs.dense_shapes[i], &dense)); in __anon952516500302()
195 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon952516500402()
216 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon952516500402()
290 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon952516500502()
310 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon952516500502()
Dfunctional_ops.cc128 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon22ceab0f0202()
157 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon22ceab0f0302()
217 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon22ceab0f0502()
287 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon22ceab0f0702()
Dtpu_outfeed_ops.cc62 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon71ff5a600102()
Dtpu_infeed_ops.cc60 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon9e8f715f0102()
Dresource_variable_ops.cc111 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon3f9cfa090202()
144 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &s)); in __anon3f9cfa090302()
Dlookup_ops.cc403 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in __anon8a0811960c02()
438 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in __anon8a0811960d02()
Dstate_ops.cc52 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anone3ce2a1d0102()
Ddataset_ops.cc534 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in IteratorGetNextShapeFn()
714 c->MakeShapeFromPartialTensorShape(s, &output_shape_handle)); in __anondee5e0a21402()
Dexperimental_dataset_ops.cc551 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in GetShapeFn()
Darray_ops.cc290 c->MakeShapeFromPartialTensorShape(shape, &passed_shape)); in __anon7c94107b0202()
1456 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon7c94107b2302()
1611 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(final_shape, &out)); in __anon7c94107b2502()
1883 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon7c94107b2d02()
1909 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon7c94107b2e02()
Ddata_flow_ops.cc584 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon4a7ac08d1102()
736 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon4a7ac08d1602()
Dlist_ops.cc286 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(raw_element_shape, in __anonebbd57d80702()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc78 TF_RETURN_IF_ERROR(context->MakeShapeFromPartialTensorShape( in PropagateShapes()
89 context->MakeShapeFromPartialTensorShape(arg_shape.shape, &handle)); in PropagateShapes()
/external/tensorflow/tensorflow/contrib/framework/ops/
Dvariable_ops.cc55 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anonf00e47a70202()
/external/tensorflow/tensorflow/contrib/image/ops/
Dsingle_image_random_dot_stereograms_ops.cc54 c->MakeShapeFromPartialTensorShape(shape, &output_image_shape)); in __anonbda156eb0102()
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
Darray_ops.cc37 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon96b295990102()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc101 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext()
112 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext()
130 MakeShapeFromPartialTensorShape(p.first, &new_v[j].shape)); in InferenceContext()
918 Status InferenceContext::MakeShapeFromPartialTensorShape( in MakeShapeFromPartialTensorShape() function in tensorflow::shape_inference::InferenceContext
936 return MakeShapeFromPartialTensorShape(PartialTensorShape(shape.dim_sizes()), in MakeShapeFromTensorShape()
945 return MakeShapeFromPartialTensorShape(partial_shape, out); in MakeShapeFromShapeProto()
Dshape_inference_test.cc1126 TEST_F(ShapeInferenceTest, MakeShapeFromPartialTensorShape) { in TEST_F() argument
1133 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape(PartialTensorShape(), &out)); in TEST_F()
1138 c.MakeShapeFromPartialTensorShape(PartialTensorShape({0}), &out)); in TEST_F()
1140 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape( in TEST_F()
1612 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TestMergeHandles()
1723 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TestRelaxHandles()
Dshape_inference.h492 Status MakeShapeFromPartialTensorShape(
Dcommon_shape_fns.cc1610 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &output_shape)); in ExplicitShape()
1624 c->MakeShapeFromPartialTensorShape(shapes[i], &output_shape)); in ExplicitShapes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1976 ic->MakeShapeFromPartialTensorShape(shape, &shape_handle)); in UpdateQueue()