Searched refs:max_array_size_ (Results 1 – 2 of 2) sorted by relevance
74 max_array_size_(max_array_size), in XlaResource()83 xla::XlaOp(), max_array_size_, /*tensor_array_gradients=*/{}, in XlaResource()138 ta_shape.AddDim(max_array_size_); in SetZeroValue()146 ta_shape.AddDim(max_array_size_); in SetZeroValue()171 ta_shape.AddDim(max_array_size_); in GetOrCreateTensorArrayGradient()178 type_, shape_, gradient_value, max_array_size_, in GetOrCreateTensorArrayGradient()
148 int64 max_array_size() const { return max_array_size_; } in max_array_size()149 void set_max_array_size(int64 size) { max_array_size_ = size; } in set_max_array_size()180 int64 max_array_size_ = -1; variable