Home
last modified time | relevance | path

Searched refs:inferred_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/operators/
Ddata_structures.py78 inferred_shape, = tuple(all_shapes)
79 if element_shape is not None and element_shape != inferred_shape:
82 element_shape, elements, inferred_shape))
89 inferred_shape = None
94 element_shape = inferred_shape
138 inferred_shape = array_ops.shape(elements[0])
139 if element_shape is not None and element_shape != inferred_shape:
142 element_shape, elements, inferred_shape))
149 inferred_shape = constant_op.constant(-1) # unknown shape, by convention
151 inferred_shape = constant_op.constant(-1) # unknown shape, by convention
[all …]
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py262 inferred_shape = self.as_tensor_shape(
265 inferred_shape[0]))
266 self.assertEqual(test_case['shape'][1:], inferred_shape[1:])
291 inferred_shape = self.as_tensor_shape(
294 inferred_shape[0]))
295 self.assertEqual(test_case['shape'][1:], inferred_shape[1:])
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc430 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
432 inferred_shape)); in TEST_F()
476 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
478 inferred_shape)); in TEST_F()
522 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
524 inferred_shape)); in TEST_F()
1436 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
1438 ShapeUtil::Equal(ShapeUtil::MakeShape(F32, {39, 31}), inferred_shape)); in TEST_F()
1454 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
1455 ASSERT_TRUE(ShapeUtil::Equal(input_shape, inferred_shape)); in TEST_F()
[all …]
Dshape_inference.cc2682 Shape inferred_shape = in InferReshapeShape() local
2685 << ShapeUtil::HumanString(inferred_shape); in InferReshapeShape()
2687 if (ShapeUtil::ElementsIn(operand) != ShapeUtil::ElementsIn(inferred_shape)) { in InferReshapeShape()
2692 ShapeUtil::ElementsIn(inferred_shape), in InferReshapeShape()
2693 ShapeUtil::HumanString(inferred_shape)); in InferReshapeShape()
2708 ShapeUtil::DimensionsUnmodifiedByReshape(operand, inferred_shape); in InferReshapeShape()
2711 inferred_shape.set_dynamic_dimension(unmodified.second, true); in InferReshapeShape()
2715 return inferred_shape; in InferReshapeShape()
2792 Shape inferred_shape = ShapeUtil::ChangeElementType( in InferSelectShape() local
2797 for (int i = 0; i < inferred_shape.rank(); i++) { in InferSelectShape()
[all …]
Dhlo_verifier.cc847 const Shape& inferred_shape) { in CheckShape() argument
881 return ShapesSame(instruction->shape(), inferred_shape); in CheckShape()
889 inferred_shape); in CheckShape()
891 return ShapeUtil::Compatible(instruction->shape(), inferred_shape); in CheckShape()
899 StringifyShape(inferred_shape), StringifyShape(instruction->shape()), in CheckShape()
Dhlo_verifier.h109 const Shape& inferred_shape);
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc326 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation() argument
327 return inferred_shape.shape; in PerformStaticShapeInferenceBeforeEncapsulation()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc988 bool CompatibleShapes(ShapeHandle inferred_shape, in CompatibleShapes() argument
990 if (inferred_shape.SameHandle(annotated_shape)) { in CompatibleShapes()
993 if (!InferenceContext::RankKnown(inferred_shape)) { in CompatibleShapes()
996 if (InferenceContext::Rank(inferred_shape) != in CompatibleShapes()
1000 const int rank = InferenceContext::Rank(inferred_shape); in CompatibleShapes()
1002 if (!InferenceContext::DimKnownRank(inferred_shape, i) in CompatibleShapes()
1006 InferenceContext::DimKnownRank(inferred_shape, i)); in CompatibleShapes()