Home
last modified time | relevance | path

Searched refs:shape_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc293 TensorShape shape_attr; in __anon3500048a0202() local
294 TF_RETURN_IF_ERROR(c->GetAttr("shape", &shape_attr)); in __anon3500048a0202()
296 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(shape_attr, &s)); in __anon3500048a0202()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_test.cc145 tensorflow::AttrValue shape_attr; in TestPreserveOutputShapes() local
146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes()
153 placeholder.node()->AddAttr("_output_shapes", shape_attr); in TestPreserveOutputShapes()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc1978 AttrValue shape_attr; in CreateIdentityGraphDef() local
1979 shape_attr.mutable_shape()->set_unknown_rank(true); in CreateIdentityGraphDef()
1985 placeholder->mutable_attr()->insert({"shape", shape_attr}); in CreateIdentityGraphDef()