/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A 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() 310 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anonc6fc979c0302()
|
H A D | tpu_outfeed_ops.cc | 63 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon9bead88d0102() 97 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon9bead88d0202()
|
H A D | resource_variable_ops.cc | 44 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in ReadVariableShapeFn() 108 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon9faba5f60202() 141 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &s)); in __anon9faba5f60302()
|
H A D | tpu_infeed_ops.cc | 60 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon42bd5dec0102()
|
H A D | lookup_ops.cc | 347 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in MutableHashTableOfTensorsShapeFn() 355 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(value_p, &value_s)); in MutableDenseHashTableShapeFn()
|
H A D | state_ops.cc | 52 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon9ff5ca8a0102()
|
H A D | array_ops.cc | 297 c->MakeShapeFromPartialTensorShape(shape, &passed_shape)); in __anon38bbb0e80202() 1618 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon38bbb0e82302() 1783 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(final_shape, &out)); in __anon38bbb0e82502() 2071 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon38bbb0e82d02() 2097 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon38bbb0e82e02()
|
H A D | parsing_ops.cc | 37 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(dense_shape, &s)); in AddDenseOutputShapes()
|
H A D | data_flow_ops.cc | 639 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon1926cf7a1502() 791 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon1926cf7a1a02()
|
H A D | list_ops.cc | 311 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(raw_element_shape, in __anon5e0c1ea50702()
|
H A D | dataset_ops.cc | 1160 c->MakeShapeFromPartialTensorShape(s, &output_shape_handle)); in __anon4377504f2102()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | shape_inference.cc | 47 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext() 58 construction_status_.Update(MakeShapeFromPartialTensorShape(p, &shape)); in InferenceContext() 76 MakeShapeFromPartialTensorShape(p.first, &new_v[j].shape)); in InferenceContext() 882 Status InferenceContext::MakeShapeFromPartialTensorShape( in MakeShapeFromPartialTensorShape() function in tensorflow::shape_inference::InferenceContext 900 return MakeShapeFromPartialTensorShape(PartialTensorShape(shape.dim_sizes()), in MakeShapeFromTensorShape() 922 return MakeShapeFromPartialTensorShape(partial_shape, out); in MakeShapeFromShapeProto()
|
H A D | shape_inference_test.cc | 1170 TEST_F(ShapeInferenceTest, MakeShapeFromPartialTensorShape) { in TEST_F() argument 1177 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape(PartialTensorShape(), &out)); in TEST_F() 1182 c.MakeShapeFromPartialTensorShape(PartialTensorShape({0}), &out)); in TEST_F() 1184 TF_ASSERT_OK(c.MakeShapeFromPartialTensorShape( in TEST_F() 1711 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TEST_P() 1815 TF_CHECK_OK(c.MakeShapeFromPartialTensorShape(S(dim_sizes), &s)); in TEST_P()
|
H A D | shape_inference.h | 563 Status MakeShapeFromPartialTensorShape(
|
H A D | common_shape_fns.cc | 490 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in DatasetIteratorShape() 2482 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &output_shape)); in ExplicitShape() 2496 c->MakeShapeFromPartialTensorShape(shapes[i], &output_shape)); in ExplicitShapes()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
H A D | shape_inference.cc | 89 TF_RETURN_IF_ERROR(context->MakeShapeFromPartialTensorShape( in PropagateShapes() 100 context->MakeShapeFromPartialTensorShape(arg_shape.shape, &handle)); in PropagateShapes()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ops/ |
H A D | trt_engine_op.cc | 55 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon9f7885b30102()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/ |
H A D | aot_only_var_handle_op.cc | 74 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon74f606920202()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
H A D | xla_ops.cc | 1297 c->MakeShapeFromPartialTensorShape(shapes_attr[i], &s)); in __anon6b9e6b801302()
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
H A D | dtensor_device_util.cc | 737 TF_RETURN_IF_ERROR(inference_context->MakeShapeFromPartialTensorShape( in PrepareGraphForMlir()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
H A D | graph_properties.cc | 2453 ic->MakeShapeFromPartialTensorShape(shape, &shape_handle)); in UpdateQueue()
|