OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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