Searched refs:kVariable (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 295 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in ReadVariableInput()
327 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in GetVariableTypeAndShape()
420 TF_RET_CHECK(variable->kind() == XlaResource::kVariable); in AssignVariable()
|
| D | xla_resource.h | 34 kVariable, enumerator
|
| D | xla_resource.cc | 93 case kVariable: { in SetZeroValue()
|
| D | xla_compiler_test.cc | 707 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
767 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
|
| D | xla_compiler.cc | 244 case XlaResource::kVariable: { in XLAShapeForArgument()
|
| /external/swiftshader/third_party/subzero/src/ |
| D | IceOperand.h | 53 kVariable, enumerator
702 Variable(Func, kVariable, Ty, Index); in create()
852 return Kind >= kVariable && Kind <= kVariable_Max; in classof()
1156 BooleanVariable(Func, kVariable, Ty, Index); in create()
|
| D | IceOperand.cpp | 209 Variable(Func, kVariable, Ty, Number); in asType()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 186 arg.resource_kind = XlaResource::kVariable; in BuildArguments()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 42 constexpr char kVariable[] = "Variable"; variable
180 {kVariable, wrap(&OpLevelCostEstimator::PredictVariable)}, in OpLevelCostEstimator()
|