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