Searched refs:output_tensors_as_shapes (Results 1 – 1 of 1) sorted by relevance
570 std::vector<ShapeHandle> output_tensors_as_shapes; member693 ctx->output_tensors_as_shapes.resize(grappler_function_item.output_size()); in UpdateFunction()731 &ctx->output_tensors_as_shapes[output]); in UpdateFunction()785 if (src_ctx->output_tensors_as_shapes.size() > src_output) { in UpdateNode()787 src_ctx->output_tensors_as_shapes[src_output]; in UpdateNode()1133 if (c->output_tensors_as_shapes.size() < ic->num_outputs() && in AllOutputValuesKnown()1144 if (c->output_tensors_as_shapes.size() > i && in AllOutputValuesKnown()1145 ic->FullyDefined(c->output_tensors_as_shapes[i])) { in AllOutputValuesKnown()1293 c->output_tensors_as_shapes.resize(outputs.size()); in UpdateOutputShapesAndValues()1312 MaybeTensorValueToShape(ic, *t.tensor, &c->output_tensors_as_shapes[k]); in UpdateOutputShapesAndValues()[all …]