| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops.cc | 106 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in IfShapeInferenceFn()
151 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in CaseShapeInferenceFn()
223 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in WhileShapeInferenceFn()
309 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anonb31eee6d0302()
|
| D | tpu_outfeed_ops.cc | 63 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon0c58097e0102()
97 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon0c58097e0202()
|
| D | tpu_infeed_ops.cc | 60 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon2edfb4bd0102()
|
| D | resource_variable_ops.cc | 93 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon50d007a70202()
126 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &s)); in __anon50d007a70302()
|
| D | lookup_ops.cc | 403 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in __anon18b155f40c02()
438 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in __anon18b155f40d02()
|
| D | state_ops.cc | 52 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anonaa115abb0102()
|
| D | parsing_ops.cc | 37 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(dense_shape, &s)); in AddDenseOutputShapes()
|
| D | data_flow_ops.cc | 639 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anonea01102b1502()
791 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anonea01102b1a02()
|
| D | array_ops.cc | 296 c->MakeShapeFromPartialTensorShape(shape, &passed_shape)); in __anon42d741190202()
1595 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon42d741192302()
1750 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(final_shape, &out)); in __anon42d741192502()
2038 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon42d741192d02()
2064 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon42d741192e02()
|
| D | list_ops.cc | 297 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(raw_element_shape, in __anonabedecb60702()
|
| D | dataset_ops.cc | 969 c->MakeShapeFromPartialTensorShape(s, &output_shape_handle)); in __anon42b7b0c02002()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | aot_only_var_handle_op.cc | 74 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon5de01ee30202()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 89 TF_RETURN_IF_ERROR(context->MakeShapeFromPartialTensorShape( in PropagateShapes()
100 context->MakeShapeFromPartialTensorShape(arg_shape.shape, &handle)); in PropagateShapes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 46 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext()
57 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext()
75 MakeShapeFromPartialTensorShape(p.first, &new_v[j].shape)); in InferenceContext()
862 Status InferenceContext::MakeShapeFromPartialTensorShape( in MakeShapeFromPartialTensorShape() function in tensorflow::shape_inference::InferenceContext
880 return MakeShapeFromPartialTensorShape(PartialTensorShape(shape.dim_sizes()), in MakeShapeFromTensorShape()
889 return MakeShapeFromPartialTensorShape(partial_shape, out); in MakeShapeFromShapeProto()
|
| D | shape_inference_test.cc | 1136 TEST_F(ShapeInferenceTest, MakeShapeFromPartialTensorShape) { in TEST_F() argument
1143 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape(PartialTensorShape(), &out)); in TEST_F()
1148 c.MakeShapeFromPartialTensorShape(PartialTensorShape({0}), &out)); in TEST_F()
1150 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape( in TEST_F()
1611 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TestMergeHandles()
1722 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TestRelaxHandles()
|
| D | shape_inference.h | 478 Status MakeShapeFromPartialTensorShape(
|
| D | common_shape_fns.cc | 487 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in DatasetIteratorShape()
2390 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &output_shape)); in ExplicitShape()
2404 c->MakeShapeFromPartialTensorShape(shapes[i], &output_shape)); in ExplicitShapes()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2409 ic->MakeShapeFromPartialTensorShape(shape, &shape_handle)); in UpdateQueue()
|