HomeSort by relevance Sort by last modified time
    Searched full:custom_call (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc 317 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) {
324 if (custom_call->custom_call_target() ==
326 const HloInstruction* epsilon = custom_call->operand(5);
330 const HloInstruction* feature_index = custom_call->operand(6);
336 /*operand=*/GetAllocationSlice(*custom_call->operand(0)),
337 /*scale=*/GetAllocationSlice(*custom_call->operand(1)),
338 /*offset=*/GetAllocationSlice(*custom_call->operand(2)),
339 /*mean=*/GetAllocationSlice(*custom_call->operand(3)),
340 /*variance=*/GetAllocationSlice(*custom_call->operand(4)),
343 /*output=*/GetAllocationSlice(*custom_call),
    [all...]
ir_emission_utils.cc 153 HloInstruction* custom_call = local
156 custom_call->set_window(window);
157 custom_call->set_convolution_dimension_numbers(dnums);
158 return custom_call;
cudnn_convolution_rewriter.cc 399 HloInstruction* custom_call = [&]() -> HloInstruction* { local
432 if (custom_call == nullptr) {
440 HloInstruction::CreateGetTupleElement(conv->shape(), custom_call, 0)));
cudnn_convolution_rewriter_test.cc 288 const HloInstruction* custom_call = local
291 const WindowDimension& window_dim = custom_call->window().dimensions(i);
417 const HloInstruction* custom_call = local
420 const WindowDimension& window_dim = custom_call->window().dimensions(i);
ir_emitter_unnested.h 66 Status HandleCustomCall(HloInstruction* custom_call) override;
ir_emitter.h 93 Status HandleCustomCall(HloInstruction* custom_call) override;
  /external/tensorflow/tensorflow/compiler/xla/service/
dfs_hlo_visitor_with_default.h 130 Status HandleCustomCall(HloInstructionPtr custom_call) override {
131 return DefaultAction(custom_call);
hlo_element_type_converter_test.cc 37 HloModule custom_call
hlo_cost_analysis.h 87 Status HandleCustomCall(const HloInstruction* custom_call) override;
layout_assignment.cc 674 Status CheckCustomCallLayout(HloInstruction* custom_call) {
675 for (const HloInstruction* operand : custom_call->operands()) {
681 ShapeUtil::IsOpaque(custom_call->shape()) ||
682 LayoutUtil::IsMonotonicWithDim0Major(custom_call->shape().layout()));
    [all...]
buffer_assignment_test.cc 1201 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc     [all...]
ir_emitter.h 149 Status HandleCustomCall(HloInstruction* custom_call) override;
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser_test.cc 832 R"(HloModule custom_call
    [all...]

Completed in 412 milliseconds