OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:OperandIndices
(Results
1 - 13
of
13
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding.h
28
using
OperandIndices
= std::vector<int64>;
32
using FoldableOperands = std::function<
OperandIndices
(const HloInstruction&,
33
const
OperandIndices
&)>;
38
static
OperandIndices
NeverFoldTranspose(const HloInstruction&,
39
const
OperandIndices
&) {
transpose_folding.cc
34
TransposeFolding::
OperandIndices
CanFoldOperandsIntoDot(
42
TransposeFolding::
OperandIndices
operand_set;
53
TransposeFolding::
OperandIndices
CanFoldOperandsIntoConvolution(
61
TransposeFolding::
OperandIndices
operand_set;
73
std::pair<HloInstruction*, TransposeFolding::
OperandIndices
>;
179
std::vector<std::pair<HloInstruction*,
OperandIndices
>> foldable_dots;
180
std::vector<std::pair<HloInstruction*,
OperandIndices
>> foldable_convolutions;
184
OperandIndices
operand_indices =
191
OperandIndices
operand_indices = CanFoldOperandsIntoConvolution(
liveness_util.cc
123
for (int64 op_idx : alias_user->
OperandIndices
(alias.instruction())) {
143
if (fusion->
OperandIndices
(operand).size() > 1) {
243
std::vector<int64> operand_indices = user->
OperandIndices
(operand);
250
const std::vector<int64> operand_indices = user->
OperandIndices
(operand);
347
std::vector<int64> operand_indices = user->
OperandIndices
(operand);
hlo_value.cc
179
for (int64 operand_number : user->
OperandIndices
(position.instruction)) {
transpose_folding_test.cc
45
const TransposeFolding::
OperandIndices
& candidate_operands) {
49
const TransposeFolding::
OperandIndices
& candidate_operands) {
hlo_dataflow_analysis.cc
646
for (int64 operand_number : user->
OperandIndices
(instruction)) {
[
all
...]
hlo_instruction.h
[
all
...]
algebraic_simplifier.cc
[
all
...]
layout_assignment.cc
[
all
...]
hlo_instruction.cc
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.cc
183
const TransposeFolding::
OperandIndices
& candidate_operands) {
185
: TransposeFolding::
OperandIndices
{};
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc
281
const TransposeFolding::
OperandIndices
& candidate_operands) {
284
: TransposeFolding::
OperandIndices
{};
[
all
...]
cpu_instruction_fusion_test.cc
180
const TransposeFolding::
OperandIndices
& candidate_operands) {
Completed in 1282 milliseconds