HomeSort by relevance Sort by last modified time
    Searched defs:XlaResource (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_resource.cc 28 XlaResource::XlaResource(Kind kind, int arg_num, string name, DataType type,
45 new XlaResource(/*kind=*/kTensorArray, /*arg_num=*/-1,
52 Status XlaResource::SetTypeAndShape(DataType type, const TensorShape& shape) {
76 Status XlaResource::SetValue(const xla::ComputationDataHandle& value) {
86 Status XlaResource::SetZeroValue(xla::ComputationBuilder* builder) {
124 Status XlaResource::GetOrCreateTensorArrayGradient(
126 XlaResource** gradient_out) {
130 std::unique_ptr<XlaResource>& gradient = tensor_array_gradients_[source];
138 new XlaResource(/*kind=*/kTensorArray, /*arg_num=*/-1
    [all...]
xla_resource.h 30 class XlaResource {
39 XlaResource(Kind kind, int arg_num, string name, DataType type,
45 XlaResource(const XlaResource&) = delete;
46 XlaResource(XlaResource&&) = delete;
47 XlaResource& operator=(const XlaResource&) = delete;
48 XlaResource& operator=(XlaResource&&) = delete
    [all...]

Completed in 4964 milliseconds