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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_fused_conv_rewriter.cc 146 CastOrNull<HloBroadcastInstruction>(bias_broadcast_instr);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_instructions.cc     [all...]
hlo_instructions.h 555 class HloBroadcastInstruction : public HloInstruction {
557 explicit HloBroadcastInstruction(const Shape& shape, HloInstruction* operand,
    [all...]
hlo_instruction.cc     [all...]

Completed in 856 milliseconds