Searched defs:inferred_shape (Results 1 – 10 of 10) sorted by relevance
50 const Shape& inferred_shape = status_or_inferred_shape.ValueOrDie().value(); in InstructionMatchesPattern() local
442 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local488 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local534 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local814 Shape inferred_shape = inferred_status.ValueOrDie(); in Pass() local2085 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local2103 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local2174 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local2184 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local2396 Shape inferred_shape = statusor.ValueOrDie(); in TEST_F() local
2930 Shape inferred_shape = ShapeUtil::MakeShape( in InferDynamicReshapeShape() local2948 Shape inferred_shape = in InferReshapeShape() local
1175 const Shape& inferred_shape, in CheckShape()
336 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
653 for (const auto& inferred_shape : output_shapes) { in GetGraph() local676 for (const auto& inferred_shape : output_shapes) { in GetGraph() local
76 gtl::InlinedVector<int64, 4> inferred_shape; member
81 for (const auto& inferred_shape : output_shapes) { in ConvertGraphShapeInfoToShapeMap() local
1200 bool CompatibleShapes(ShapeHandle inferred_shape, in CompatibleShapes()1229 bool SameShapes(ShapeHandle inferred_shape, in SameShapes()
1021 const PartialTensorShape& inferred_shape, int replica_id, in CreateConcatNodesForRetval()