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.cc50 auto kTensorArray = XlaResourceKind::kTensorArray; in CreateResourceOpInfoMap() local
101 add("TensorArrayV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
102 add("TensorArrayConcatV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
103 add("TensorArrayGatherV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
104 add("TensorArrayScatterV3" , kWrite, kTensorArray); in CreateResourceOpInfoMap()
105 add("TensorArrayGradV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
106 add("TensorArrayCloseV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
107 add("TensorArrayReadV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
108 add("TensorArraySizeV3" , kRead, kTensorArray); in CreateResourceOpInfoMap()
109 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()
81 /*kind=*/kTensorArray, /*arg_num=*/-1, in XlaResource()
136 case kTensorArray: { in SetZeroValue()
167 TF_RET_CHECK(kind_ == kTensorArray); in GetOrCreateTensorArrayGradient()
176 new XlaResource(/*kind=*/kTensorArray, /*arg_num=*/-1, in GetOrCreateTensorArrayGradient()
190 TF_RET_CHECK(kind_ == kTensorArray); in Pack()
210 TF_RET_CHECK(kind_ == kTensorArray); in SetFromPack()
Dresource_operation_table.h38 kTensorArray // Operates on tensor arrays. enumerator
Dxla_resource.h36 kTensorArray, enumerator
Dresource_util.cc79 op_info->resource_kind() != XlaResourceKind::kTensorArray) in IsStackOrTensorArraySource()
Dxla_compiler.cc307 arg.resource_kind != XlaResource::kTensorArray && in BuildComputation()
886 case XlaResource::kTensorArray: { in XLAShapeForArgument()
Dxla_compiler_test.cc817 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
876 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
908 args[0].resource_kind = XlaResource::kTensorArray; in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.h249 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.cc86 if (arg.resource_kind == XlaResource::kTensorArray) { in MakeXlaCompilerArgumentsFromInputs()