Searched refs:resource_kind (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 69 if (std::tie(kind, resource_kind, type, name, initialized, tensor_array_size, in operator ==()
71 std::tie(other.kind, other.resource_kind, other.type, other.name, in operator ==()
243 switch (arg.resource_kind) { in XLAShapeForArgument()
450 TF_RET_CHECK(arg.resource_kind != XlaResource::kInvalid); in BuildArguments()
455 arg.resource_kind, i, arg.name, arg.type, arg.shape, in BuildArguments()
|
| D | xla_compiler_test.cc | 473 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
539 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
571 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
707 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
767 args[1].resource_kind = XlaResource::kVariable; in TEST_F()
|
| D | xla_compiler.h | 130 XlaResource::Kind resource_kind = XlaResource::kInvalid; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 55 arg.resource_kind = resource->kind(); in MakeXlaCompilerArgumentsFromInputs()
56 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 186 arg.resource_kind = XlaResource::kVariable; in BuildArguments()
|