OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kdef
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_gpu_backend.cc
21
bool GpuOpFilter(KernelDef*
kdef
) {
24
if (
kdef
->op() == "RandomStandardNormal" ||
kdef
->op() == "RandomUniform" ||
25
kdef
->op() == "RandomUniformInt" ||
kdef
->op() == "TruncatedNormal") {
xla_cpu_backend.cc
21
bool CpuOpFilter(KernelDef*
kdef
) {
24
if (
kdef
->op() == "RandomStandardNormal") {
25
kdef
->clear_constraint();
27
KernelDef::AttrConstraint* attr_constraint =
kdef
->add_constraint();
tf2xla_supported_ops.cc
48
for (const KernelDef*
kdef
: kdefs) {
50
for (const KernelDef::AttrConstraint& constraint :
kdef
->constraint()) {
59
std::cout << "`" <<
kdef
->op() << "` | "
xla_op_registry.cc
197
std::unique_ptr<KernelDef>
kdef
(new KernelDef);
198
kdef
->set_op(op_registration->name);
199
kdef
->set_device_type(backend.first);
206
KernelDef::AttrConstraint* attr_constraint =
kdef
->add_constraint();
242
!backend.second.op_filter(
kdef
.get())) {
249
new KernelDef(*
kdef
), "XlaJitOp", op_registration->factory));
250
backend.second.kernel_defs.push_back(std::move(
kdef
));
xla_op_registry.h
95
typedef bool (*BackendOpFilter)(KernelDef*
kdef
);
/external/tensorflow/tensorflow/compiler/jit/
xla_interpreter_device.cc
61
static bool OpFilter(KernelDef*
kdef
) { return true; }
/external/tensorflow/tensorflow/core/framework/
memory_types.cc
79
const KernelDef*
kdef
= nullptr;
local
81
FindKernelDef(device_type, ndef, &
kdef
, nullptr /* kernel_class_name */);
113
const auto& from_proto =
kdef
->host_memory_arg();
/external/tensorflow/tensorflow/core/debug/
debug_graph_utils.cc
274
const KernelDef*
kdef
;
local
310
Status s = FindKernelDef(device_type, node_def, &
kdef
, nullptr);
445
const KernelDef*
kdef
;
local
465
if (!FindKernelDef(DeviceType(device.device_type()), node_def, &
kdef
, nullptr)
/external/tensorflow/tensorflow/core/common_runtime/
constant_folding.cc
523
const KernelDef*
kdef
;
local
524
if (!FindKernelDef(device_type, def, &
kdef
, nullptr).ok()) {
Completed in 131 milliseconds