Home
last modified time | relevance | path

Searched refs:kVariable (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table.cc48 auto kVariable = XlaResourceKind::kVariable; in CreateResourceOpInfoMap() local
53 add("AssignAddVariableOp" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
54 add("AssignSubVariableOp" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
55 add("AssignVariableOp" , kWrite, kVariable); in CreateResourceOpInfoMap()
56 add("ReadVariableOp" , kRead, kVariable); in CreateResourceOpInfoMap()
57 add("ResourceApplyAdaMax" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
58 add("ResourceApplyAdadelta" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
59 add("ResourceApplyAdagrad" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
60 add("ResourceApplyAdagradDA" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
61 add("ResourceApplyAdam" , kReadWrite, kVariable); in CreateResourceOpInfoMap()
[all …]
Dresource_operation_table.h36 kVariable, // Operates on resource variables. enumerator
Dxla_resource.cc34 case XlaResource::kVariable: in KindToString()
130 case kVariable: { in SetZeroValue()
Dxla_resource.h35 kVariable, enumerator
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()
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()
1269 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
1455 args[0].resource_kind = XlaResource::kVariable; in TEST_F()
Dxla_op_kernel.cc376 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in ReadVariableInputTensor()
426 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in GetVariableTypeAndShape()
516 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in AssignVariableTensor()
Dxla_expression_test.cc45 XlaResource::kVariable, /*arg_num=*/0, /*name=*/string("avariable"), in SetUp()
Dtf2xla.cc277 arg.resource_kind = XlaResource::kVariable; in ConvertGraphToXla()
Dxla_compiler.cc643 case XlaResource::kVariable: { in XLAShapeForArgument()
/external/tensorflow/tensorflow/compiler/aot/
Daot_only_var_handle_op.cc43 if (resource->kind() == XlaResource::kVariable && in Compile()
/external/v8/tools/clang/rewrite_to_chrome_style/
DEditTracker.h30 kVariable, enumerator
DEditTracker.cpp24 case RenameCategory::kVariable: in GetTag()
DRewriteToChromeStyle.cpp1188 return RenameCategory::kVariable; in GetCategory()
/external/v8/src/torque/
Ddeclarable.h29 kVariable, enumerator
50 bool IsVariable() const { return kind() == kVariable; } in IsVariable()
151 : Value(Declarable::kVariable, type, name), in Variable()
/external/swiftshader/third_party/subzero/src/
DIceOperand.h53 kVariable, enumerator
710 Variable(Func, kVariable, Ty, Index); in create()
860 return Kind >= kVariable && Kind <= kVariable_Max; in classof()
1171 BooleanVariable(Func, kVariable, Ty, Index); in create()
DIceOperand.cpp209 Variable(Func, kVariable, Ty, Number); in asType()
/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()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc72 constexpr char kVariable[] = "Variable"; variable
285 kConst, kVariable, kVariableV2, kAutoReloadVariable, in OpLevelCostEstimator()