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

  /external/tensorflow/tensorflow/compiler/xla/service/
bfloat16_conversion_folding.h 19 #include "tensorflow/compiler/xla/service/bfloat16_support.h"
36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support)
37 : bfloat16_support_(bfloat16_support) {}
bfloat16_normalization.h 19 #include "tensorflow/compiler/xla/service/bfloat16_support.h"
30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support)
31 : bfloat16_support_(bfloat16_support) {}
bfloat16_conversion_folding.cc 32 HloComputation* computation, const BFloat16Support* bfloat16_support)
33 : computation_(computation), bfloat16_support_(bfloat16_support) {}
41 const BFloat16Support* bfloat16_support) {
42 BFloat16ConversionFoldingVisitor visitor(computation, bfloat16_support);
bfloat16_propagation.h 26 #include "tensorflow/compiler/xla/service/bfloat16_support.h"
65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);
bfloat16_normalization.cc 33 const BFloat16Support* bfloat16_support)
34 : computation_(computation), bfloat16_support_(bfloat16_support) {}
39 const BFloat16Support* bfloat16_support) {
40 BFloat16NormalizationVisitor visitor(computation, bfloat16_support);
bfloat16_propagation.cc 34 const BFloat16Support* bfloat16_support)
35 : bfloat16_support_(bfloat16_support) {}
    [all...]
bfloat16_propagation_test.cc 17 #include "tensorflow/compiler/xla/service/bfloat16_support.h"
67 TestBFloat16Support bfloat16_support; local
68 BFloat16Propagation propagation(&bfloat16_support);
    [all...]

Completed in 2995 milliseconds