Home
last modified time | relevance | path

Searched refs:GetOrCreateTensorArrayGradient (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_resource.cc165 Status XlaResource::GetOrCreateTensorArrayGradient(const string& source, in GetOrCreateTensorArrayGradient() function in tensorflow::XlaResource
224 GetOrCreateTensorArrayGradient(source, builder, &gradient)); in SetFromPack()
H A Dxla_resource.h121 Status GetOrCreateTensorArrayGradient(const string& source,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dcase_op.cc185 OP_REQUIRES_OK(ctx, resource->GetOrCreateTensorArrayGradient( in Compile()
H A Dif_op.cc74 TF_RETURN_IF_ERROR(resource->GetOrCreateTensorArrayGradient( in PopulateTensorArrayGradients()
H A Dtensor_array_ops.cc593 ctx, resource->GetOrCreateTensorArrayGradient(source_, b, &gradient)); in Compile()
H A Dwhile_op.cc405 OP_REQUIRES_OK(ctx, resource->GetOrCreateTensorArrayGradient( in Compile()