Searched refs:handle_shape (Results 1 – 7 of 7) sorted by relevance
90 arg_shape.handle_shape, &handle)); in PropagateShapes()228 &output.handle_shape)); in StoreOutputShapes()238 << output.handle_shape.DebugString(); in StoreOutputShapes()290 a.handle_shape.MergeWith(b.handle_shape, &result.handle_shape)); in MergeInferredShapes()
36 PartialTensorShape handle_shape; member
169 EXPECT_TRUE(iter->second.at(0).handle_shape.AsTensorShape(&resource_shape)); in TEST()
27 reserved 2; // was handle_shape
1717 !info->handle_shape.IsFullyDefined())) { in GetArgAndRetvalShapes()1747 (*arg_shapes)[i].handle_shape = PartialTensorShape(); in GetArgAndRetvalShapes()1773 arg_shape.handle_shape = info->handle_shape; in GetArgAndRetvalShapes()2254 : &arg_shapes[i].handle_shape; in BuildDynamicShapeNodes()2369 arg_shape.handle_shape.AsProto(arg->mutable_shape()); in BuildCompileNode()2704 if (!raw_var_shape.handle_shape.AsTensorShape(&var_shape) && in CreatePerHostDummyArgs()3173 arg_shapes[orig_arg_num].handle_shape, replica, in BuildExecuteNodes()3257 arg_shapes[orig_arg_num].handle_shape, replica, in BuildExecuteNodes()3428 arg_shapes[orig_arg_num].handle_shape, replica, orig_inputs, in BuildExecuteNodes()
835 handle_shape = tensor_shape.TensorShape(838 handle_shape)
470 shape_for_inference.handle_shape = arg_shapes[i]; in RunShapeInferenceOnComputation()