Searched refs:true_b (Results 1 – 3 of 3) sorted by relevance
1931 true_b = array_ops.slice(all_b, [0], array_ops.shape(labels_flat))1947 true_b = array_ops.reshape(true_b, [-1, num_true])1948 true_logits += true_b
591 true_w, true_b = weights[labels], biases[labels]597 true_b = true_b.reshape((batch_size, num_true))598 true_logits += true_b
2198 SpmdBuilder true_b("true_computation", visiting_hlo_); in HandleSingleDevice() local2201 auto param = true_b.AddInstruction(HloInstruction::CreateParameter( in HandleSingleDevice()2205 new_operands.push_back(true_b.AddInstruction( in HandleSingleDevice()2208 auto root = true_b.AddInstruction( in HandleSingleDevice()2210 true_computation = module_->AddEmbeddedComputation(true_b.Build(root)); in HandleSingleDevice()