Searched defs:outputTensor (Results 1 – 4 of 4) sorted by relevance
37 TensorStr4D outputTensor; member
400 EigenTensor outputTensor(output_shape_, output); in LaunchKernel() local450 EigenTensor outputTensor(output_shape_, output); in LaunchKernel() local
396 EigenTensor outputTensor(output, output->GetData()); in ScaleAndTranslateCompute() local461 EigenTensor outputTensor(output, output->GetData()); in ScaleAndTranslateGradCompute() local
465 repeated Tensor outputTensor = 5; // must be the same length as the "output" field field