Home
last modified time | relevance | path

Searched refs:refined_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc486 SmallVectorImpl<int64_t> *refined_shape) { in GetCastCompatibleShape() argument
489 refined_shape->reserve(rank); in GetCastCompatibleShape()
495 refined_shape->push_back(dim2); in GetCastCompatibleShape()
499 refined_shape->push_back(dim1); in GetCastCompatibleShape()
503 refined_shape->push_back(dim1); in GetCastCompatibleShape()
796 SmallVector<int64_t, 4> refined_shape; in GetCastCompatibleType() local
797 if (!GetCastCompatibleShape(a_tt.getShape(), b_tt.getShape(), &refined_shape)) in GetCastCompatibleType()
800 return RankedTensorType::get(refined_shape, refined_element_ty); in GetCastCompatibleType()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc116 bool refined_shape = false; in TypeMeet() local
125 refined_shape = true; in TypeMeet()
133 refined_shape = true; in TypeMeet()
188 if (refined_shape || lhs_element_type != lhs_shape_type.getElementType()) { in TypeMeet()