Searched refs:max_array_size_ (Results 1 – 2 of 2) sorted by relevance
75 max_array_size_(max_array_size), in XlaResource()84 xla::XlaOp(), max_array_size_, /*tensor_array_gradients=*/{}, in XlaResource()137 ta_shape.AddDim(max_array_size_); in SetZeroValue()145 ta_shape.AddDim(max_array_size_); in SetZeroValue()170 ta_shape.AddDim(max_array_size_); in GetOrCreateTensorArrayGradient()177 type_, shape_, gradient_value, max_array_size_, in GetOrCreateTensorArrayGradient()
146 int64 max_array_size() const { return max_array_size_; } in max_array_size()147 void set_max_array_size(int64 size) { max_array_size_ = size; } in set_max_array_size()178 int64 max_array_size_ = -1; variable