Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dfunctional_ops.cc106 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 Dtpu_outfeed_ops.cc63 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 Dresource_variable_ops.cc44 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 Dtpu_infeed_ops.cc60 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon42bd5dec0102()
H A Dlookup_ops.cc347 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 Dstate_ops.cc52 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shape, &out)); in __anon9ff5ca8a0102()
H A Darray_ops.cc297 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 Dparsing_ops.cc37 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(dense_shape, &s)); in AddDenseOutputShapes()
H A Ddata_flow_ops.cc639 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon1926cf7a1502()
791 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon1926cf7a1a02()
H A Dlist_ops.cc311 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(raw_element_shape, in __anon5e0c1ea50702()
H A Ddataset_ops.cc1160 c->MakeShapeFromPartialTensorShape(s, &output_shape_handle)); in __anon4377504f2102()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc47 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 Dshape_inference_test.cc1170 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 Dshape_inference.h563 Status MakeShapeFromPartialTensorShape(
H A Dcommon_shape_fns.cc490 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 Dshape_inference.cc89 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 Dtrt_engine_op.cc55 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape( in __anon9f7885b30102()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/
H A Daot_only_var_handle_op.cc74 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(p, &s)); in __anon74f606920202()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/
H A Dxla_ops.cc1297 c->MakeShapeFromPartialTensorShape(shapes_attr[i], &s)); in __anon6b9e6b801302()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc737 TF_RETURN_IF_ERROR(inference_context->MakeShapeFromPartialTensorShape( in PrepareGraphForMlir()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc2453 ic->MakeShapeFromPartialTensorShape(shape, &shape_handle)); in UpdateQueue()