Searched defs:output_shards (Results 1 – 3 of 3) sorted by relevance
337 std::vector<Tensor> output_shards(num_batches * num_returns); in batchedTensorForLoopFallback() local467 std::vector<Tensor> output_shards(num_components * num_returns); in batchedNestedTensorForLoopFallback() local
321 std::vector<Tensor> output_shards(num_batches * num_returns); in batchedTensorForLoopFallback() local
400 std::vector<int32> output_shards = output_layout->num_shards(); in ExpandOp() local