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

  /external/tensorflow/tensorflow/compiler/xla/service/
dfs_hlo_visitor_with_default.h 164 Status HandleReduce(HloInstructionPtr reduce) override {
hlo_cost_analysis.h 86 Status HandleReduce(const HloInstruction* reduce) override;
hlo_evaluator.h 239 Status HandleReduce(HloInstruction* reduce) override;
dynamic_dimension_inference.cc 44 Status HandleReduce(HloInstruction* hlo) override;
166 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) {
dfs_hlo_visitor.h 228 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
hlo_verifier.h 69 Status HandleReduce(HloInstruction* reduce) override;
hlo_cost_analysis.cc 278 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) {
hlo_evaluator.cc     [all...]
hlo_verifier.cc 441 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) {
    [all...]
algebraic_simplifier.cc 228 Status HandleReduce(HloInstruction* reduce) override;
    [all...]
hlo_instruction_test.cc 97 Status HandleReduce(HloInstruction* reduce) override {
    [all...]
hlo_evaluator_typed_visitor.h     [all...]
hlo_instruction.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.h 92 Status HandleReduce(HloInstruction* reduce) override;
ir_emitter_unnested.h 169 Status HandleReduce(HloInstruction* reduce) override;
ir_emitter.cc 664 Status IrEmitter::HandleReduce(HloInstruction* reduce) {
    [all...]
ir_emitter_unnested.cc 722 Status IrEmitterUnnested::HandleReduce(HloInstruction* reduce) {
731 return IrEmitter::HandleReduce(reduce);
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
tile_ops.cc 376 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \
421 void HandleReduce(OpKernelContext* context,
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.h 162 Status HandleReduce(HloInstruction* reduce) override;
ir_emitter.cc     [all...]

Completed in 1296 milliseconds