Searched refs:CheckShape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 119 return CheckShape(concatenate, in HandleConcatenate()
125 return CheckShape(convert, ShapeInference::InferConvertShape( in HandleConvert()
131 return CheckShape(convert, ShapeInference::InferBitcastConvertShape( in HandleBitcastConvert()
147 return CheckShape(dot, expected); in HandleDot()
158 return CheckShape(convolution, expected); in HandleConvolution()
166 return CheckShape(fft, expected); in HandleFft()
174 return CheckShape(hlo, expected); in HandleTriangularSolve()
181 return CheckShape(hlo, expected); in HandleCholesky()
275 return CheckShape(ag, ShapeInference::InferAllGatherShape( in HandleAllGather()
288 return CheckShape(hlo, ShapeInference::InferAllReduceShape(operand_shapes)); in HandleAllReduce()
[all …]
|
| D | hlo_verifier.h | 120 Status CheckShape(const HloInstruction* instruction,
125 Status CheckShape(const HloInstruction* instruction,
|
| /external/llvm-project/flang/lib/Semantics/ |
| D | assignment.cpp | 48 void CheckShape(parser::CharBlock, const SomeExpr *);
79 CheckShape(lhsLoc, &lhs); in Analyze()
220 void AssignmentContext::CheckShape(parser::CharBlock at, const SomeExpr *expr) { in CheckShape() function in Fortran::semantics::AssignmentContext
249 CheckShape(expr.thing.value().source, GetExpr(expr)); in PushWhereContext()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 74 XlaOp CheckShape(XlaBuilder* b, const XlaOp& operand, in CheckShape() function in xla::__anoncfc9dfd00111::BatchNormalizationTest
176 CheckShape(&builder, ConstantLiteral(&builder, input_literal_), in XLA_TEST_P()
183 auto sum = CheckShape( in XLA_TEST_P()
202 CheckShape(&builder, in XLA_TEST_P()
209 CheckShape(&builder, Gt(standard_deviation, epsilon), in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tflite_driver.cc | 213 bool CheckShape(bool verbose, const TfLiteTensor& tensor) { in CheckShape() function in tflite::testing::TfLiteDriver::ShapeExpectation
674 if (!p.second->CheckShape(/*verbose=*/false, *tensor)) { in CheckResults()
680 p.second->CheckShape(/*verbose=*/true, *tensor); in CheckResults()
|