Home
last modified time | relevance | path

Searched refs:resource_kind (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table.h44 XlaResourceKind resource_kind) in XlaResourceOpInfo() argument
45 : op_kind_(op_kind), resource_kind_(resource_kind) {} in XlaResourceOpInfo()
48 XlaResourceKind resource_kind() const { return resource_kind_; } in resource_kind() function
Dresource_operation_table.cc38 XlaResourceKind resource_kind) { in CreateResourceOpInfoMap() argument
40 result->insert({op, XlaResourceOpInfo(op_kind, resource_kind)}); in CreateResourceOpInfoMap()
Dxla_compiler_test.cc305 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
356 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
399 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
748 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
808 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
841 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
1003 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
1106 args[0].resource_kind = XlaResource::kVariable; in TEST_F()
1157 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
1199 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
[all …]
Dxla_compiler.cc344 if (std::tie(kind, resource_kind, type, name, initialized, max_array_size, in operator ==()
346 std::tie(other.kind, other.resource_kind, other.type, other.name, in operator ==()
388 XlaResource::KindToString(resource_kind), in HumanString()
642 switch (arg.resource_kind) { in XLAShapeForArgument()
724 TF_RET_CHECK(arg.resource_kind != XlaResource::kInvalid); in BuildArguments()
730 arg.resource_kind, i, arg.name, arg.type, in BuildArguments()
Dxla_compiler.h148 XlaResource::Kind resource_kind = XlaResource::kInvalid; member
Dtf2xla.cc277 arg.resource_kind = XlaResource::kVariable; in ConvertGraphToXla()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc64 arg.resource_kind = resource->kind(); in MakeXlaCompilerArgumentsFromInputs()
65 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()
Dif_op.cc70 arg.resource_kind = resource->kind(); in Compile()
Dcase_op.cc67 arg.resource_kind = resource->kind(); in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc160 return op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in IsNonResourceVarResourceOp()
600 op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in FindCompilationCandidates()
Dxla_launch_util.cc478 arg.resource_kind = XlaResource::kVariable; in BuildXlaCompilerArguments()