| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_verifier.h | 137 bool SameElementType(const Shape& a, const Shape& b) {
140 : ShapeUtil::SameElementType(a, b);
|
| hlo_verifier.cc | 275 return ShapeUtil::SameElementType(shape_0, shape_1) &&
276 (ShapeUtil::SameElementType(shape_0, result_shape) ||
430 if (!ShapeUtil::SameElementType(parameter_shape, operand_shape)) {
480 TF_RET_CHECK(SameElementType(broadcast->shape(), operand_shape));
497 TF_RET_CHECK(SameElementType(reshape->shape(), operand_shape));
[all...] |
| hlo_evaluator_typed_visitor.h | [all...] |
| algebraic_simplifier.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.h | 250 static bool SameElementType(const Shape& lhs, const Shape& rhs) {
254 // As SameElementType, but allows floating point types to have different
261 return ShapeUtil::SameElementType(a, b);
269 if (SameElementType(a, b)) {
[all...] |
| shape.cc | 138 (!ignore_fp_precision_ && !ShapeUtil::SameElementType(lhs, rhs))) {
|
| shape_util.cc | [all...] |
| literal.cc | 511 TF_RET_CHECK(ShapeUtil::SameElementType(src_literal.shape(), shape()));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | [all...] |
