Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table.cc51 auto kTensorArray = XlaResourceKind::kTensorArray; in CreateResourceOpInfoMap() local
105 add("TensorArrayV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
106 add("TensorArrayConcatV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
107 add("TensorArrayGatherV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
108 add("TensorArrayScatterV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
109 add("TensorArrayGradV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
110 add("TensorArrayCloseV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
111 add("TensorArrayReadV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
112 add("TensorArraySizeV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
113 add("TensorArraySplitV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
[all …]
Dxla_resource.cc37 case XlaResource::kTensorArray: in KindToString()
56 XlaResource::kTensorArray, /*arg_num=*/-1, std::move(name), type, shape, in CreateTensorArray()
83 /*kind=*/kTensorArray, /*arg_num=*/-1, in XlaResource()
139 case kTensorArray: { in SetZeroValue()
170 TF_RET_CHECK(kind_ == kTensorArray); in GetOrCreateTensorArrayGradient()
179 new XlaResource(/*kind=*/kTensorArray, /*arg_num=*/-1, in GetOrCreateTensorArrayGradient()
193 TF_RET_CHECK(kind_ == kTensorArray); in Pack()
213 TF_RET_CHECK(kind_ == kTensorArray); in SetFromPack()
Dresource_operation_table.h38 kTensorArray // Operates on tensor arrays. enumerator
Dxla_resource.h37 kTensorArray, enumerator
Dresource_util.cc79 op_info->resource_kind() != XlaResourceKind::kTensorArray) in IsStackOrTensorArraySource()
Dxla_compiler.cc311 arg.resource_kind != XlaResource::kTensorArray && in BuildComputation()
912 case XlaResource::kTensorArray: { in XLAShapeForArgument()
Dxla_compiler_test.cc823 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
881 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
913 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.h245 return op_info && op_info->resource_kind() == XlaResourceKind::kTensorArray; in IsTensorArrayOp()
/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.cc87 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()