Home
last modified time | relevance | path

Searched refs:resource_kind (Results 1 – 13 of 13) 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
Dxla_argument.cc23 if (std::tie(kind, resource_kind, type, name, initialized, max_array_size, in operator ==()
25 std::tie(other.kind, other.resource_kind, other.type, other.name, in operator ==()
Dresource_operation_table.cc38 XlaResourceKind resource_kind) { in CreateResourceOpInfoMap() argument
40 result->insert({op, XlaResourceOpInfo(op_kind, resource_kind)}); in CreateResourceOpInfoMap()
Dxla_argument.h89 XlaResource::Kind resource_kind = XlaResource::kInvalid; member
Dxla_compiler_test.cc300 args[0].resource_kind = XlaResource::kVariable; in TEST_F()
341 args[0].resource_kind = XlaResource::kVariable; in TEST_F()
396 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
449 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
512 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
817 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
876 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
908 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
1068 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
1175 args[0].resource_kind = XlaResource::kVariable; in TEST_F()
[all …]
Dresource_util.cc78 if (op_info->resource_kind() != XlaResourceKind::kStack && in IsStackOrTensorArraySource()
79 op_info->resource_kind() != XlaResourceKind::kTensorArray) in IsStackOrTensorArraySource()
Dxla_compiler.cc307 arg.resource_kind != XlaResource::kTensorArray && in BuildComputation()
454 " resource_kind=", XlaResource::KindToString(resource_kind), in HumanString()
874 switch (arg.resource_kind) { in XLAShapeForArgument()
940 arg->resource_kind = resource.kind(); in PopulateArgumentFromResource()
976 TF_RET_CHECK(arg.resource_kind != XlaResource::kInvalid); in BuildArguments()
982 arg.resource_kind, i, arg.name, arg.type, in BuildArguments()
Dgraph_compiler_util.cc259 arg.resource_kind = XlaResource::kVariable; in PopulateXlaArgs()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.h243 return op_info && op_info->resource_kind() == XlaResourceKind::kStack; in IsStackOp()
249 return op_info && op_info->resource_kind() == XlaResourceKind::kTensorArray; in IsTensorArrayOp()
Dxla_launch_util.cc620 arg.resource_kind = XlaResource::kVariable; in BuildXlaCompilerArguments()
Dmark_for_compilation_pass.cc1269 op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in FindCompilationCandidates()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc86 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc201 arg.resource_kind = XlaResource::kVariable; in BuildComputationArgumentDescriptions()