Home
last modified time | relevance | path

Searched defs:inferred_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc430 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
476 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
522 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
1436 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
1454 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
1525 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local
1535 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local
1747 Shape inferred_shape = statusor.ValueOrDie(); in TEST_F() local
Dshape_inference.cc2682 Shape inferred_shape = in InferReshapeShape() local
2792 Shape inferred_shape = ShapeUtil::ChangeElementType( in InferSelectShape() local
Dhlo_verifier.cc847 const Shape& inferred_shape) { in CheckShape()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc326 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc988 bool CompatibleShapes(ShapeHandle inferred_shape, in CompatibleShapes()