Home
last modified time | relevance | path

Searched refs:branch_b (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_subclassing_test_util.py50 branch_b = [shared_layer]
52 branch_b.append(keras.layers.BatchNormalization())
53 branch_b.append(keras.layers.Dense(num_classes[1], activation='softmax'))
57 branch_a, branch_b, name='test_model'))
Dtemporal_sample_weights_correctness_test.py54 branch_b = [inp_2, x, out_2]
55 return testing_utils.get_multi_io_model(branch_a, branch_b)
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py670 def __init__(self, branch_a, branch_b, shared_input_branch=None, argument
675 self._branch_b = branch_b
753 branch_b, argument
838 inputs = branch_a[0], branch_b[0]
840 branch_b = branch_b[1:]
844 return _MultiIOSubclassModel(branch_a, branch_b, shared_input_branch,
849 (lambda: branch_b),
869 for layer in branch_b:
Dmetrics_correctness_test.py44 branch_b = [inp_2, x, out_2]
45 return testing_utils.get_multi_io_model(branch_a, branch_b)
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2567 SpmdBuilder branch_b(absl::StrCat("infeed_branch_", i), visiting_hlo_); in HandleInfeed() local
2568 auto param = branch_b.AddInstruction(HloInstruction::CreateParameter( in HandleInfeed()
2570 auto infeed = branch_b.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed()
2590 branch_b.AddInstruction(HloInstruction::CreateGetTupleElement( in HandleInfeed()
2594 return branch_b.AddInstruction( in HandleInfeed()
2604 return PadToShape(infeed_element, pad_shape, &branch_b); in HandleInfeed()
2608 return CreateZero(pad_shape, &branch_b); in HandleInfeed()
2612 branches[i] = module_->AddEmbeddedComputation(branch_b.Build()); in HandleInfeed()
2971 SpmdBuilder branch_b(absl::StrCat("outfeed_branch_", i), visiting_hlo_); in HandleOutfeed() local
2973 auto param = branch_b.AddInstruction(HloInstruction::CreateParameter( in HandleOutfeed()
[all …]
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py125 branch_b = [input_b, dense, dense2, dropout]
127 model = testing_utils.get_multi_io_model(branch_a, branch_b)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_dataset_test.py163 branch_b = [input_b, dense, dropout]
165 model = testing_utils.get_multi_io_model(branch_a, branch_b)
Dtraining_test.py283 branch_b = [input_b, dense, dropout]
285 model = testing_utils.get_multi_io_model(branch_a, branch_b)
1938 branch_b = [input_b, dense, dropout]
1940 model = testing_utils.get_multi_io_model(branch_a, branch_b)
2914 branch_b = [inp_2, x, out_2]
2915 model = testing_utils.get_multi_io_model(branch_a, branch_b)