| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | concat_builder.cc | 33 int concat_axis = concat_params->axis; in PopulateSubGraph() local
36 concat_axis = in PopulateSubGraph()
37 concat_axis < 0 ? concat_axis + 4 : concat_axis + 4 - output_dim_size; in PopulateSubGraph()
39 kScalarShape, reinterpret_cast<char*>(&concat_axis), sizeof(concat_axis)); in PopulateSubGraph()
|
| /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/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/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/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 173 int64_t concat_axis; in convertPackOp() local
177 concat_axis = 0; in convertPackOp()
181 concat_axis = 0; in convertPackOp()
195 concat_axis = axis; in convertPackOp()
199 IntegerAttr concat_axis_attr = rewriter.getI64IntegerAttr(concat_axis); in convertPackOp()
214 orig_input_dim_on_axis = input0_tensor_shape[concat_axis]; in convertPackOp()
217 concat_output_shape[concat_axis] = orig_input_dim_on_axis * 2; in convertPackOp()
226 concat_output_shape[concat_axis] = orig_input_dim_on_axis * (i + 1); in convertPackOp()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 393 int32 concat_axis = split ? 3 : 2; in TestFoldFusedBatchNormsWithConcat() local
394 test::FillValues<int32>(&shape_tensor, {concat_axis}); in TestFoldFusedBatchNormsWithConcat()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 525 size_t concat_axis = positive_axis(axis)
541 output_shape[concat_axis] = output_shape[concat_axis] * %input.size
544 …%concat = tosa.CONCAT(%input[0], %input[1]) {axis=concat_axis} : (tensor<%input[0].type>, tensor<%…
548 …%concat = tosa.CONCAT(%concat, %input[i]) {axis=concat_axis} : (tensor<%concat.type>, tensor<%inpu…
558 if (concat_axis == input_tensor_rank) {
561 concat_axis = 0
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 2745 const Input& concat_axis, Output* output) { in CreateConcatV1Op() argument
2753 auto axis = ops::AsNodeOut(scope, concat_axis); in CreateConcatV1Op()
|