| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_concat_op_test.py | 144 for concat_dim in (-2000, 1, 2000):
145 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a])
166 for concat_dim in (-1, 1):
167 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
192 for concat_dim in (-2, 0):
193 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_d])
217 for concat_dim in (-1, 1):
218 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b, sp_c])
241 for concat_dim in (-1, 1):
242 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
[all …]
|
| D | concat_op_test.py | 118 concat_dim = np.random.randint(5)
128 input_shape[concat_dim] = np.random.randint(1, 5)
139 c = array_ops.concat(concat_inputs, concat_dim)
152 ind[concat_dim] = slice(cur_offset,
153 cur_offset + params[p[i]].shape[concat_dim])
154 cur_offset += params[p[i]].shape[concat_dim]
287 concat_dim = np.random.randint(5)
294 shape[concat_dim] = x
300 c = array_ops.concat(inp_tensors, concat_dim)
302 output_shape[concat_dim] = concat_dim_sizes.sum()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_concat_op.cc | 109 const int32 concat_dim) { in CalculateInputsDim() argument
111 for (int d = 0; d < concat_dim; ++d) { in CalculateInputsDim()
119 OpKernelContext* context, const int32 concat_dim, in CalculateConcatDims() argument
141 if (j == concat_dim) { in CalculateConcatDims()
156 *output_concat_dim += in.dims() > 0 ? in.dim_size(concat_dim) : 1; in CalculateConcatDims()
168 const int32 concat_dim = concat_dim_tensor->scalar<int32>()(); in Compute() local
187 context, (0 <= concat_dim && concat_dim < input_dims), in Compute()
190 ", ", input_dims, "), but got ", concat_dim)); in Compute()
198 const int64 inputs_flat_dim0 = CalculateInputsDim(input_shape, concat_dim); in Compute()
201 CalculateConcatDims(N, input_shape, input_dims, values, context, concat_dim, in Compute()
[all …]
|
| D | concat_op.cc | 74 int64 concat_dim; in Compute() local
91 concat_dim = in Compute()
94 concat_dim = in Compute()
103 int32 axis = concat_dim < 0 ? concat_dim + input_dims : concat_dim; in Compute()
105 OP_REQUIRES(c, (0 <= axis && axis < input_dims) || concat_dim == 0, in Compute()
109 -input_dims, ", ", input_dims, "), but got ", concat_dim)); in Compute()
250 const Tensor& concat_dim = ctx->input(0); in Compute() local
252 ctx, TensorShapeUtils::IsScalar(concat_dim.shape()), in Compute()
255 concat_dim.shape().DebugString())); in Compute()
284 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); in Compute()
|
| D | sparse_concat_op.cc | 78 const int concat_dim = (concat_dim_attr_ < 0) in Compute() local
81 OP_REQUIRES(context, concat_dim >= 0 && concat_dim < input_rank, in Compute()
93 if (j != concat_dim) { in Compute()
117 concat_order.push_back(concat_dim); in Compute()
119 if (j != concat_dim) { in Compute()
|
| D | concat_op_test.cc | 67 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatHelper() local
68 concat_dim.scalar<int32>()() = concat_dimension; in ConcatHelper()
78 .Input(test::graph::Constant(g, concat_dim)) in ConcatHelper()
165 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatManyHelper() local
166 concat_dim.scalar<int32>()() = concat_dimension; in ConcatManyHelper()
177 .Input(test::graph::Constant(g, concat_dim)) in ConcatManyHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 52 int64 concat_dim; in Compile() local
54 ctx->ConstantInputAsIntScalar(axis_index_, &concat_dim)); in Compile()
63 int32 axis = concat_dim < 0 ? concat_dim + input_dims : concat_dim; in Compile()
68 -input_dims, ", ", input_dims, "), but got ", concat_dim)); in Compile()
92 VLOG(1) << "Concat dim " << concat_dim << " equivalent to " << axis; in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Concat.pbtxt | 5 name: "concat_dim"
15 and their sizes must match in all dimensions except `concat_dim`.
22 `concat_dim` dimension. This tensor's shape matches that of `values` except
23 in `concat_dim` where it has the sum of the sizes.
|
| D | api_def_QuantizedConcat.pbtxt | 4 name: "concat_dim"
14 and their sizes must match in all dimensions except `concat_dim`.
33 `concat_dim` dimension. This tensor's shape matches that of `values` except
34 in `concat_dim` where it has the sum of the sizes.
|
| D | api_def_ConcatV2.pbtxt | 10 and their sizes must match in all dimensions except `concat_dim`.
24 `concat_dim` dimension. This tensor's shape matches that of `values` except
25 in `concat_dim` where it has the sum of the sizes.
|
| D | api_def_SparseConcat.pbtxt | 40 name: "concat_dim"
65 For example, if `concat_dim = 1` and the inputs are
|
| D | api_def_ConcatOffset.pbtxt | 5 name: "concat_dim"
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | concat_ops_test.py | 86 concat_dim = np.random.randint(5)
96 input_shape[concat_dim] = np.random.randint(1, 5)
108 c = array_ops.concat(concat_inputs, concat_dim)
121 ind[concat_dim] = slice(cur_offset,
122 cur_offset + params[p[i]].shape[concat_dim])
123 cur_offset += params[p[i]].shape[concat_dim]
222 concat_dim = np.random.randint(5)
230 shape[concat_dim] = x
238 c = array_ops.concat(inp_tensors, concat_dim)
240 output_shape[concat_dim] = concat_dim_sizes.sum()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 147 const int32 concat_dim = in Compute() local
154 int32 axis = (concat_dim < 0) ? (concat_dim + input_dims) : concat_dim; in Compute()
159 -input_dims, ", ", input_dims, "), but got ", concat_dim)); in Compute()
473 int32 concat_dim = in Compute() local
487 if (concat_dim < 0) concat_dim = expected_dims + concat_dim; in Compute()
502 if (d == concat_dim) continue; in Compute()
603 FindMklCommonFormat(mkl_input_shapes, concat_dim, in Compute()
639 dst_concat_dim_size += src_dims[concat_dim]; in Compute()
661 dst_dims[concat_dim] = dst_concat_dim_size; in Compute()
713 concat_dim = mkl_input_shapes[0].TfDimIdx(concat_dim); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 86 concat_dim=i, values=[left_padding, input_, right_padding])
132 concat_dim=i, values=[left_offset, reversed_left_padding])
134 concat_dim=i, values=[reversed_right_padding, right_offset])
137 concat_dim=i, values=[reversed_left_padding, left_offset])
139 concat_dim=i, values=[right_offset, reversed_right_padding])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 72 def _CreateDenseMaskAndBegin(sizes, concat_dim): argument
80 array_ops.fill(array_ops.expand_dims(concat_dim, 0), 0), [1],
81 array_ops.fill(shape_of_shape - concat_dim - 1, 0)
109 concat_dim = op.inputs[dim_index]
114 if context.executing_eagerly() or isinstance(concat_dim, ops.EagerTensor):
118 concat_dim._numpy().item(0) % input_values[0]._rank()) # pylint: disable=protected-access
124 if constant_op.is_constant(concat_dim):
132 dim_context = control_flow_util.GetOutputContext(concat_dim.op)
134 value = tensor_util.constant_value(concat_dim)
135 concat_dim = constant_op.constant(value=value, dtype=concat_dim.dtype)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | constant-fold.mlir | 171 %concat_dim = constant dense<1> : tensor<i32>
180 …%offset:3 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1, %shape2) : (tensor<i32>, tensor<3xi32…
188 %concat_dim = constant dense<1> : tensor<i32>
198 …%offset:4 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1, %shape2, %shape3) : (tensor<i32>, ten…
206 %concat_dim = constant dense<-1> : tensor<i32>
215 …%offset:3 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1, %shape2) : (tensor<i32>, tensor<3xi32…
222 func @testConcatOffsetNonConstConcatDim(%concat_dim: tensor<i32>) -> (tensor<3xi32>, tensor<3xi32>)…
227 …%offset:2 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1) : (tensor<i32>, tensor<3xi32>, tensor…
234 %concat_dim = constant dense<1> : tensor<i32>
238 …%offset:2 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1) : (tensor<i32>, tensor<3xi32>, tensor…
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ConcatOffset.pbtxt | 4 name: "concat_dim"
|
| D | Concat.pbtxt | 4 name: "concat_dim"
|
| D | QuantizedConcat.pbtxt | 4 name: "concat_dim"
|
| D | SparseConcat.pbtxt | 31 name: "concat_dim"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ConcatOffset.pbtxt | 4 name: "concat_dim"
|
| D | Concat.pbtxt | 4 name: "concat_dim"
|
| D | SparseConcat.pbtxt | 31 name: "concat_dim"
|
| D | QuantizedConcat.pbtxt | 4 name: "concat_dim"
|