Home
last modified time | relevance | path

Searched refs:kTensorArray (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table.cc50 auto kTensorArray = XlaResourceKind::kTensorArray; in CreateResourceOpInfoMap() local
96 add("TensorArrayV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
97 add("TensorArrayConcatV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
98 add("TensorArrayGatherV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
99 add("TensorArrayScatterV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
100 add("TensorArrayGradV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
101 add("TensorArrayCloseV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
102 add("TensorArrayReadV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
103 add("TensorArraySizeV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
104 add("TensorArraySplitV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
[all …]
Dxla_resource.cc38 case XlaResource::kTensorArray: in KindToString()
57 XlaResource::kTensorArray, /*arg_num=*/-1, std::move(name), type, shape, in CreateTensorArray()
82 /*kind=*/kTensorArray, /*arg_num=*/-1, in XlaResource()
135 case kTensorArray: { in SetZeroValue()
166 TF_RET_CHECK(kind_ == kTensorArray); in GetOrCreateTensorArrayGradient()
175 new XlaResource(/*kind=*/kTensorArray, /*arg_num=*/-1, in GetOrCreateTensorArrayGradient()
189 TF_RET_CHECK(kind_ == kTensorArray); in Pack()
209 TF_RET_CHECK(kind_ == kTensorArray); in SetFromPack()
Dresource_operation_table.h38 kTensorArray // Operates on tensor arrays. enumerator
Dxla_resource.h36 kTensorArray, enumerator
Dxla_compiler_test.cc748 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()
Dxla_compiler.cc651 case XlaResource::kTensorArray: { in XLAShapeForArgument()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_array_ops.cc53 if (resource->kind() != XlaResource::kTensorArray) { in MaybeInitializeTensorArray()
96 if (resource->kind() != XlaResource::kTensorArray) { in CheckTensorArrayIsInitialized()
Dwhile_op.cc65 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()