Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc73 HloEvaluator evaluator; in TEST_F()
92 HloEvaluator evaluator; in TEST_F()
115 HloEvaluator evaluator; in TEST_F()
134 HloEvaluator evaluator; in TEST_F()
153 HloEvaluator evaluator; in TEST_F()
172 HloEvaluator evaluator; in TEST_F()
194 HloEvaluator evaluator; in TEST_F()
213 HloEvaluator evaluator; in TEST_F()
232 HloEvaluator evaluator; in TEST_F()
254 HloEvaluator evaluator; in TEST_F()
[all …]
Dhlo_evaluator.cc178 HloEvaluator::HloEvaluator(int64 max_loop_iterations) in HloEvaluator() function in xla::HloEvaluator
232 StatusOr<Literal> HloEvaluator::Evaluate( in Evaluate()
287 StatusOr<Literal> HloEvaluator::Evaluate(HloInstruction* instruction) { in Evaluate()
315 bool HloEvaluator::TryEvaluate(HloInstruction* instruction, Literal* result) { in TryEvaluate()
327 StatusOr<Literal> HloEvaluator::EvaluateWithSubstitutions( in EvaluateWithSubstitutions()
355 StatusOr<Literal> HloEvaluator::EvaluateElementwiseBinaryOp( in EvaluateElementwiseBinaryOp()
370 StatusOr<Literal> HloEvaluator::EvaluateElementwiseUnaryOp( in EvaluateElementwiseUnaryOp()
382 StatusOr<Literal> HloEvaluator::EvaluateDotOp( in EvaluateDotOp()
402 Status HloEvaluator::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
412 Status HloEvaluator::HandleGetDimensionSize( in HandleGetDimensionSize()
[all …]
Dhlo_evaluator.h45 class HloEvaluator : public DfsHloVisitorWithDefault {
49 explicit HloEvaluator(int64 max_loop_iterations = -1);
374 TF_DISALLOW_COPY_AND_ASSIGN(HloEvaluator);
Dhlo_constant_folding.cc70 auto evaluator = absl::make_unique<HloEvaluator>(/*max_loop_iterations=*/0); in Run()
Dhlo_evaluator_test.cc197 HloEvaluator evaluator_;
2542 HloEvaluator hlo_eval; in TEST_F()
2579 HloEvaluator hlo_eval; in BM_ReducePrecisely()
3255 HloEvaluator evaluator; in TEST_P()
3282 HloEvaluator evaluator; in TEST_P()
4302 EXPECT_EQ(HloEvaluator() in TEST_F()
4308 EXPECT_EQ(HloEvaluator() in TEST_F()
4330 HloEvaluator().Evaluate(*m_, {&input, &input}).status().error_message(), in TEST_F()
4332 EXPECT_EQ(HloEvaluator() in TEST_F()
4419 EXPECT_EQ(HloEvaluator().Evaluate(*m_, {&args[0]}).status().code(), in TEST_F()
[all …]
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()
Dwhile_loop_analysis.cc488 HloEvaluator evaluator(/*max_loop_iterations=*/0); in ComputeWhileLoopTripCount()
607 HloEvaluator evaluator(/*max_loop_iterations=*/0); in ComputeWhileLoopTripCountUpperBound()
Dhlo_evaluator_typed_visitor.h152 explicit HloEvaluatorTypedVisitor(HloEvaluator* p) : parent_(p) {}
215 (HloEvaluator::ElementWiseUnaryOpImpl<typename NativeT::value_type,
1380 HloEvaluator::MatmulArray2D(lhs_array, rhs_array);
1733 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);
1871 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);
1985 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);
2364 HloEvaluator embedded_evaluator;
2964 (HloEvaluator::ElementWiseUnaryOpImpl<ReturnT, ReturnT>(
3030 HloEvaluator* parent_;
Dservice.cc1080 HloEvaluator evaluator; in ComputeConstantGraph()
Dalgebraic_simplifier.cc2595 HloEvaluator evaluator; in MinMaxToClamp()
/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.h48 std::unique_ptr<HloEvaluator> evaluator,
60 std::unique_ptr<HloEvaluator> evaluator_ TF_PT_GUARDED_BY(evaluator_lock_);
Dexecutable.cc43 std::unique_ptr<HloEvaluator> evaluator, in InterpreterExecutable()
Dcompiler.cc114 auto evaluator = absl::make_unique<HloEvaluator>(); in RunBackend()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h61 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
Dreference_util.cc487 HloEvaluator evaluator; in ConvArray4DGeneralDimensionsDilated()