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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.h 19 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
37 const TargetMachineFeatures* target_machine_features)
38 : target_machine_features_(*target_machine_features) {}
cpu_layout_assignment.h 20 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
35 const TargetMachineFeatures* target_machine_features)
38 target_machine_features_(*target_machine_features) {}
ir_emission_utils.h 20 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
28 const TargetMachineFeatures& target_machine_features);
33 const Shape& shape, const TargetMachineFeatures& target_machine_features);
dot_op_emitter.h 22 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
37 const TargetMachineFeatures& target_machine_features);
43 const TargetMachineFeatures& target_machine_features);
68 const TargetMachineFeatures& target_machine_features);
ir_emission_utils_test.cc 43 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features(
48 *conv_instr, target_machine_features));
ir_emission_utils.cc 28 const Shape& shape, const TargetMachineFeatures& target_machine_features) {
39 return target_machine_features.minimum_alignment_for_allocation(
45 const TargetMachineFeatures& target_machine_features) {
58 return GetMinimumAlignmentForArray(shape, target_machine_features) >=
parallel_task_assignment.h 19 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
44 const TargetMachineFeatures* target_machine_features);
70 const TargetMachineFeatures* target_machine_features)
73 target_machine_features_(*target_machine_features) {}
dot_op_emitter_internal.h 18 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
83 const TargetMachineFeatures& target_machine_features);
conv_canonicalization_test.cc 94 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features(
98 ConvCanonicalization conv_canonicalization(&target_machine_features);
156 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features(
160 ConvCanonicalization conv_canonicalization(&target_machine_features);
dot_op_emitter.cc 28 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
104 const TargetMachineFeatures& target_machine_features);
117 const TargetMachineFeatures& target_machine_features);
213 const TargetMachineFeatures& target_machine_features)
223 target_machine_features_(target_machine_features) {}
753 const TargetMachineFeatures& target_machine_features) {
772 const TargetMachineFeatures& target_machine_features) {
779 dot_info.result_shape, target_machine_features);
784 const TargetMachineFeatures& target_machine_features) {
785 CHECK(IsAlignedGemm(dot_info, target_machine_features));
858 target_machine_features); local
    [all...]
cpu_layout_assignment.cc 98 const TargetMachineFeatures& target_machine_features) {
101 target_machine_features);
104 target_machine_features);
cpu_compiler.h 24 #include "tensorflow/compiler/xla/service/cpu/target_machine_features.h"
164 LLVMTargetMachineFeatures* target_machine_features);
169 LLVMTargetMachineFeatures* target_machine_features);
cpu_compiler.cc 249 LLVMTargetMachineFeatures* target_machine_features) {
280 pipeline.AddPass<ConvCanonicalization>(target_machine_features);
316 *target_machine_features)
325 LayoutAssignment::InstructionCanChangeLayout, target_machine_features);
339 LLVMTargetMachineFeatures* target_machine_features) {
379 max_parallelism, ShapeSizeBytesFunction(), target_machine_features);
396 LLVMTargetMachineFeatures target_machine_features(target_machine);
398 &target_machine_features));
400 &target_machine_features);
663 LLVMTargetMachineFeatures target_machine_features(jit->target_machine())
    [all...]
parallel_task_assignment.cc 110 const TargetMachineFeatures* target_machine_features)
111 : target_machine_features_(*target_machine_features) {
cpu_layout_assignment_test.cc 53 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features(
59 &target_machine_features);
321 cpu::TargetMachineFeaturesWithFakeAlignmentLogic target_machine_features(
327 &target_machine_features);
ir_emitter.cc 88 const TargetMachineFeatures* target_machine_features,
99 target_machine_features_(*target_machine_features),
    [all...]

Completed in 114 milliseconds