Searched refs:p2_tensor (Results 1 – 1 of 1) sorted by relevance
1372 Tensor *p1_tensor = nullptr, *p2_tensor = nullptr; in AllocateTFOutputs() local1379 AllocateOutputSetMklShape(context, kP2Index, &p2_tensor, TensorShape({}), in AllocateTFOutputs()1381 std::fill_n(p2_tensor->flat<U>().data(), p2_tensor->shape().num_elements(), in AllocateTFOutputs()