Searched refs:evaluator_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 46 evaluator_(std::move(evaluator)), in InterpreterExecutable()
49 evaluator_->set_dynamic_dimension_inference( in InterpreterExecutable()
59 evaluator_->ResetVisitStates(); in Evaluate()
60 return evaluator_->Evaluate(computation, arg_literals); in Evaluate()
|
| D | executable.h | 57 std::unique_ptr<HloEvaluator> evaluator_ ABSL_PT_GUARDED_BY(evaluator_lock_);
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionSycl.h | 139 …FullReductionKernelFunctor(LocalAccessor scratch_, Evaluator evaluator_, EvaluatorPointerType fina…
141 …: scratch(scratch_), evaluator(evaluator_), final_output(final_output_), rng(rng_), op(OpDef::get_…
222 …GenericNondeterministicReducer(Scratch, Evaluator evaluator_, EvaluatorPointerType output_accessor…
224 : evaluator(evaluator_),
270 …PartialReductionKernel(ScratchAcc scratch_, Evaluator evaluator_, EvaluatorPointerType output_acce…
274 evaluator(evaluator_),
|
| D | TensorExecutor.h | 630 const Scratch, Evaluator evaluator_, const Index range_)
631 : evaluator(evaluator_), range(range_) {}
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.h | 105 HloEvaluator evaluator_; variable
|
| D | value_inference.cc | 1606 evaluator_, in AnalyzeIsDynamic()
1669 return HloProtoEvaluator(evaluator_, *inst) in SimplifyOp()
1774 evaluator_, in AnalyzeConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 65 return evaluator_.Evaluate(*m_->entry_computation(), arg_literals);
76 return evaluator_.Evaluate(*module->entry_computation(), arg_literals)
133 TF_ASSERT_OK_AND_ASSIGN(Literal result, evaluator_.Evaluate(instruction)); in TestEvaluateInstruction()
138 StatusOr<Literal> result = evaluator_.Evaluate(instruction); in TestEvaluationFailure()
146 evaluator_.Evaluate( in TestRecursivelyEvaluateInstruction()
153 StatusOr<Literal> result = evaluator_.Evaluate( in TestRecursiveEvaluationFailure()
222 HloEvaluator evaluator_; member in xla::__anonc6f6d3400111::HloEvaluatorTest
4378 evaluator_.set_dynamic_dimension_inference(&dynamic_dimension_inference); in TEST_F()
|