Searched refs:concat_axis (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 65 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.float32
66 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.uint8
67 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int32
68 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int64
69 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.float32
70 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.uint8
71 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int32
72 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int64
73 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.float32
74 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.uint8
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_trivial_tile_to_concat.cc | 54 int concat_axis = 0; in Run() local
58 concat_axis = i; in Run()
70 LogName(*tile_op), concat_axis); in Run()
76 for (int i = 0; i < multiples[concat_axis]; ++i) { in Run()
79 concat_op->axis = concat_axis; in Run()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 223 auto concat_axis = ops::Const(host_scope.WithOpName("concat_axis"), 0); in ComputeSliceSize() local
225 host_scope.graph()->AddControlEdge(e->src(), concat_axis.node()); in ComputeSliceSize()
228 concat_axis); in ComputeSliceSize()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 392 int32 concat_axis = split ? 3 : 2; in TestFoldFusedBatchNormsWithConcat() local
393 test::FillValues<int32>(&shape_tensor, {concat_axis}); in TestFoldFusedBatchNormsWithConcat()
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 201 concat_axis = core.concat_axes(concat_axis_list)
205 concat_axes = (values[:concat_dimension] + [concat_axis] +
|