HomeSort by relevance Sort by last modified time
    Searched refs:HloEvaluator (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executable.h 41 // Responsible for running a HLO graph through the HloEvaluator and output
46 std::unique_ptr<HloEvaluator> evaluator);
62 // The interpreter interprets executables with an HloEvaluator.
63 std::unique_ptr<HloEvaluator> evaluator_ PT_GUARDED_BY(evaluator_lock_);
executable.cc 41 std::unique_ptr<HloEvaluator> evaluator)
96 // Execute the graph using the HloEvaluator.
compiler.cc 120 // In this case we are using an HloEvaluator at execution time, so we don't
123 auto evaluator = absl::make_unique<HloEvaluator>();
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_creation_utils_test.cc 58 HloEvaluator evaluator;
77 HloEvaluator evaluator;
100 HloEvaluator evaluator;
119 HloEvaluator evaluator;
138 HloEvaluator evaluator;
157 HloEvaluator evaluator;
179 HloEvaluator evaluator;
198 HloEvaluator evaluator;
217 HloEvaluator evaluator;
hlo_evaluator.cc 172 // the non-typed HloEvaluator (parent evaluator) would not support them either
176 HloEvaluator::HloEvaluator(int64 max_loop_iterations)
230 StatusOr<Literal> HloEvaluator::Evaluate(
235 2, "HloEvaluator::Evaluate computation:\n" + computation.ToString());
277 StatusOr<Literal> HloEvaluator::Evaluate(HloInstruction* instruction) {
296 bool HloEvaluator::TryEvaluate(HloInstruction* instruction, Literal* result) {
308 StatusOr<Literal> HloEvaluator::EvaluateWithSubstitutions(
336 StatusOr<Literal> HloEvaluator::EvaluateElementwiseBinaryOp(
351 StatusOr<Literal> HloEvaluator::EvaluateElementwiseUnaryOp
    [all...]
hlo_constant_folding.cc 68 // retains the behavior from before while loop support in HloEvaluator and may
70 auto evaluator = absl::make_unique<HloEvaluator>(/*max_loop_iterations=*/0);
hlo_evaluator.h 43 class HloEvaluator : public DfsHloVisitorWithDefault {
47 explicit HloEvaluator(int64 max_loop_iterations = -1);
350 TF_DISALLOW_COPY_AND_ASSIGN(HloEvaluator);
indexed_array_analysis.cc     [all...]
while_loop_analysis.cc 346 HloEvaluator evaluator(/*max_loop_iterations=*/0);
465 HloEvaluator evaluator(/*max_loop_iterations=*/0);
hlo_evaluator_test.cc 54 // Test fixture for the HloEvaluator.
134 HloEvaluator evaluator_;
150 // Verifies that HloEvaluator evaluates a HLO instruction that performs clamp
221 // Verifies that HloEvaluator evaluates a HLO instruction that performs select
246 // Verifies that HloEvaluator evaluates a HLO instruction that performs
255 // Verifies that HloEvaluator evaluates a HLO instruction that performs
264 // Verifies that HloEvaluator evaluates a HLO instruction that performs
273 // Verifies that HloEvaluator evaluates a HLO instruction that performs
282 // Verifies that HloEvaluator evaluates a HLO instruction that performs
293 // Verifies that HloEvaluator evaluates a HLO instruction that perform
    [all...]
hlo_evaluator_typed_visitor.h 67 // Templated DfsHloVisitor for use by HloEvaluator.
76 // These operations are handled outside of the parent HloEvaluator handlers
83 // This a logically a private part of HloEvaluator. It lives in this header
119 explicit HloEvaluatorTypedVisitor(HloEvaluator* p) : parent_(p) {}
148 return Unimplemented("unhandled HLO ops for HloEvaluator: %s.",
182 (HloEvaluator::ElementWiseUnaryOpImpl<float, NativeT>(
    [all...]
service.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.cc 37 return HloEvaluator::MatmulArray2D(lhs, rhs);
42 return HloEvaluator::MatmulArray2D(lhs, rhs);
47 return HloEvaluator::MatmulArray2D(lhs, rhs);
552 HloEvaluator evaluator;

Completed in 248 milliseconds