Home
last modified time | relevance | path

Searched refs:HloEvaluator (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc58 HloEvaluator evaluator; in TEST_F()
77 HloEvaluator evaluator; in TEST_F()
100 HloEvaluator evaluator; in TEST_F()
119 HloEvaluator evaluator; in TEST_F()
138 HloEvaluator evaluator; in TEST_F()
157 HloEvaluator evaluator; in TEST_F()
179 HloEvaluator evaluator; in TEST_F()
198 HloEvaluator evaluator; in TEST_F()
217 HloEvaluator evaluator; in TEST_F()
Dhlo_evaluator.cc176 HloEvaluator::HloEvaluator(int64 max_loop_iterations) in HloEvaluator() function in xla::HloEvaluator
230 StatusOr<Literal> HloEvaluator::Evaluate( in Evaluate()
277 StatusOr<Literal> HloEvaluator::Evaluate(HloInstruction* instruction) { in Evaluate()
296 bool HloEvaluator::TryEvaluate(HloInstruction* instruction, Literal* result) { in TryEvaluate()
308 StatusOr<Literal> HloEvaluator::EvaluateWithSubstitutions( in EvaluateWithSubstitutions()
336 StatusOr<Literal> HloEvaluator::EvaluateElementwiseBinaryOp( in EvaluateElementwiseBinaryOp()
351 StatusOr<Literal> HloEvaluator::EvaluateElementwiseUnaryOp( in EvaluateElementwiseUnaryOp()
363 StatusOr<Literal> HloEvaluator::EvaluateDotOp( in EvaluateDotOp()
382 Status HloEvaluator::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
392 Status HloEvaluator::HandleGetDimensionSize( in HandleGetDimensionSize()
[all …]
Dhlo_evaluator.h43 class HloEvaluator : public DfsHloVisitorWithDefault {
47 explicit HloEvaluator(int64 max_loop_iterations = -1);
350 TF_DISALLOW_COPY_AND_ASSIGN(HloEvaluator);
Dhlo_constant_folding.cc70 auto evaluator = absl::make_unique<HloEvaluator>(/*max_loop_iterations=*/0); in Run()
Dindexed_array_analysis.cc930 TakeOwnership(HloEvaluator{}.EvaluateElementwiseBinaryOp( in ComputeArrayForElementwiseBinaryOp()
935 TakeOwnership(HloEvaluator{}.EvaluateElementwiseBinaryOp( in ComputeArrayForElementwiseBinaryOp()
961 TakeOwnership(HloEvaluator{}.EvaluateElementwiseUnaryOp( in ComputeArrayForElementwiseUnaryOp()
1052 TakeOwnership(HloEvaluator{}.EvaluateDotOp( in ComputeArrayForDotWithIndexedLhs()
1088 TakeOwnership(HloEvaluator{}.EvaluateDotOp( in ComputeArrayForDotWithIndexedRhs()
Dhlo_evaluator_test.cc134 HloEvaluator evaluator_;
1456 HloEvaluator hlo_eval; in TEST_F()
1492 HloEvaluator hlo_eval; in BM_ReducePrecisely()
2052 HloEvaluator evaluator; in TEST_P()
2079 HloEvaluator evaluator; in TEST_P()
3075 EXPECT_EQ(HloEvaluator() in TEST_F()
3081 EXPECT_EQ(HloEvaluator() in TEST_F()
3103 HloEvaluator().Evaluate(*m_, {&input, &input}).status().error_message(), in TEST_F()
3105 EXPECT_EQ(HloEvaluator() in TEST_F()
3192 EXPECT_EQ(HloEvaluator().Evaluate(*m_, {&args[0]}).status().code(), in TEST_F()
[all …]
Dwhile_loop_analysis.cc346 HloEvaluator evaluator(/*max_loop_iterations=*/0); in ComputeWhileLoopTripCount()
465 HloEvaluator evaluator(/*max_loop_iterations=*/0); in ComputeWhileLoopTripCountUpperBound()
Dhlo_evaluator_typed_visitor.h119 explicit HloEvaluatorTypedVisitor(HloEvaluator* p) : parent_(p) {} in HloEvaluatorTypedVisitor()
182 (HloEvaluator::ElementWiseUnaryOpImpl<float, NativeT>( in HandleAbs()
1260 HloEvaluator::MatmulArray2D(lhs_array, rhs_array); in HandleDot()
1589 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in MapImpl()
1738 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in HandleReduce()
1887 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in HandleSelectAndScatter()
1998 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in HandleReduceWindow()
2343 HloEvaluator embedded_evaluator; in HandleScatter()
2937 (HloEvaluator::ElementWiseUnaryOpImpl<ReturnT, ReturnT>( in ElementWiseUnaryOp()
3003 HloEvaluator* parent_;
Dservice.cc1103 HloEvaluator evaluator; in ComputeConstantGraph()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
DREADME.md4 evaluating the result of the HLO graph directly with HloEvaluator, without
16 the `HloEvaluator`, allocating output buffer and finally copying evaluated
18 * [`HloEvaluator`]: traverses a HLO graph and evaluates each node in DFS
Dexecutable.h46 std::unique_ptr<HloEvaluator> evaluator);
63 std::unique_ptr<HloEvaluator> evaluator_ PT_GUARDED_BY(evaluator_lock_);
Dexecutable.cc41 std::unique_ptr<HloEvaluator> evaluator) in InterpreterExecutable()
Dcompiler.cc123 auto evaluator = absl::make_unique<HloEvaluator>(); in RunBackend()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc37 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
42 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
47 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
552 HloEvaluator evaluator; in ConvArray4DGeneralDimensionsDilated()