Searched refs:concat_offset (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ConcatOffset.pbtxt | 31 concat_offset(2, [x, y, z]) => [0, 0, 0], [0, 2, 0], [0, 5, 0]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | concat_op_test.py | 680 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
689 off = gen_array_ops.concat_offset(cdim, [s0, s1])
699 off = gen_array_ops.concat_offset(cdim, [s0, s1])
709 off = gen_array_ops.concat_offset(cdim, [s0, s1])
719 off = gen_array_ops.concat_offset(cdim, [s0, s1])
732 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
740 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
|
| /external/libtextclassifier/native/lang_id/common/ |
| D | embedding-network.cc | 155 const int concat_offset = concat_offset_[es_index]; in ConcatEmbeddings() local
167 int feature_offset = concat_offset + feature_type->base() * embedding_dim; in ConcatEmbeddings()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | concat_ops_test.py | 344 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 155 offset = gen_array_ops.concat_offset(non_neg_concat_dim, sizes)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2036 concat_offset(2, [x, y, z]) => [0, 0, 0], [0, 2, 0], [0, 5, 0]
|