Searched refs:concat_layer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge_test.py | 168 concat_layer = keras.layers.Concatenate(axis=1)
169 o = concat_layer([i1, i2])
180 self.assertEqual(concat_layer.compute_mask([i1, i2], [None, None]), None)
184 concat_layer.compute_mask(
188 unit_length_o = concat_layer([i1])
192 concat_layer.compute_mask([i1, i2], x1)
194 concat_layer.compute_mask(i1, [None, None])
196 concat_layer.compute_mask([i1, i2], [None])
199 concat_layer(i1)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | node_test.py | 93 concat_layer = DummyLayer()
95 node_module.Node(layer=concat_layer, call_args=([a_2, b_2],),
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2715 nvinfer1::IConcatenationLayer* concat_layer = network()->addConcatenation( in DynamicReshape() local
2718 SetLayerName(concat_layer, params->node_def, "concat", op_instance); in DynamicReshape()
2719 concat_layer->setAxis(0); in DynamicReshape()
2720 nvinfer1::ITensor* new_shape = concat_layer->getOutput(0); in DynamicReshape()
|