Searched defs:inferred_shape (Results 1 – 12 of 12) sorted by relevance
50 const Shape& inferred_shape = status_or_inferred_shape.ValueOrDie().value(); in InstructionMatchesPattern() local57 Shape inferred_shape = MaybeInferShape(instruction).ValueOrDie().value(); in ExpandInstruction() local
50 const Shape& inferred_shape = status_or_inferred_shape.ValueOrDie().value(); in InstructionMatchesPattern() local
468 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local514 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local560 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local838 Shape inferred_shape = inferred_status.ValueOrDie(); in Pass() local2117 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local2135 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local2206 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local2216 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local2451 Shape inferred_shape = statusor.ValueOrDie(); in TEST_F() local
3070 Shape inferred_shape = ShapeUtil::MakeShape( in InferDynamicReshapeShape() local3088 Shape inferred_shape = in InferReshapeShape() local
1701 const Shape& inferred_shape, in CheckShape()
334 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
155 for (const auto& inferred_shape : output_shapes) { in ConvertGraphShapeInfoToShapeMap() local
658 for (const auto& inferred_shape : output_shapes) { in GetGraph() local681 for (const auto& inferred_shape : output_shapes) { in GetGraph() local
76 gtl::InlinedVector<int64_t, 4> inferred_shape; member
442 InferredShape inferred_shape = {partial_tensor_shape}; in GetInputOutputInfo() local2055 InferredShape inferred_shape = {partial_tensor_shape}; in InferShapesWithResourceVar() local
1237 bool CompatibleShapes(ShapeHandle inferred_shape, in CompatibleShapes()1266 bool SameShapes(ShapeHandle inferred_shape, in SameShapes()
1234 const PartialTensorShape& inferred_shape, int replica_id, in CreateConcatNodesForRetval()3218 auto inferred_shape = arg_shapes[orig_arg_num]; in CreateOrGetPerHostVariableCopy() local