HomeSort by relevance Sort by last modified time
    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