Home
last modified time | relevance | path

Searched refs:cdim (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconcat_op.cc161 int64 cdim; in Compile() local
162 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &cdim)); in Compile()
164 VLOG(1) << "ConcatOffset " << cdim << "," << inp0_rank; in Compile()
165 int32 axis = cdim < 0 ? cdim + inp0_rank : cdim; in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dconcat_op.cc284 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); in Compute() local
286 int32 axis = cdim < 0 ? cdim + dims : cdim; in Compute()
288 errors::InvalidArgument("Concat dim is out of range: ", cdim, in Compute()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconcat_op_test.py676 cdim = constant_op.constant(1, dtypes.int32)
680 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
686 cdim = constant_op.constant(1, dtypes.int32)
689 off = gen_array_ops.concat_offset(cdim, [s0, s1])
696 cdim = constant_op.constant(4, dtypes.int32)
699 off = gen_array_ops.concat_offset(cdim, [s0, s1])
706 cdim = constant_op.constant(1, dtypes.int32)
709 off = gen_array_ops.concat_offset(cdim, [s0, s1])
716 cdim = constant_op.constant(1, dtypes.int32)
719 off = gen_array_ops.concat_offset(cdim, [s0, s1])
[all …]
/external/tremolo/Tremolo/
Dfloor1.c308 int cdim=info->klass[classv].class_dim; in floor1_inverse1() local
320 for(k=0;k<cdim;k++){ in floor1_inverse1()
330 j+=cdim; in floor1_inverse1()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h444 const int cdim = (d < primary_dim) ? d : d + 1; in Concat() local
445 DCHECK_EQ(final_shape[cdim], st_shape[cdim]) in Concat()
448 << ", mismatched shape at dim: " << cdim in Concat()
/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc1164 const int64 cdim = user.concatenate_dimension(); in GetShardingFromUser() local
1166 if (tile_assignment.dim(cdim) == 1) { in GetShardingFromUser()
1185 start_offset += op->shape().dimensions(cdim); in GetShardingFromUser()
1187 const int64 tile_shape = CeilOfRatio(user.shape().dimensions(cdim), in GetShardingFromUser()
1188 tile_assignment.dimensions()[cdim]); in GetShardingFromUser()
1191 start_indices[cdim] = start_offset / tile_shape; in GetShardingFromUser()
1192 end_indices[cdim] = CeilOfRatio( in GetShardingFromUser()
1193 start_offset + instruction.shape().dimensions(cdim), tile_shape); in GetShardingFromUser()
/external/tensorflow/tensorflow/compiler/tests/
Dconcat_ops_test.py340 cdim = constant_op.constant(1, dtypes.int32)
344 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
/external/javassist/src/test/javassist/
DJvstTest2.java739 int cdim[] = { 1, 1, 2, 1, 1, 2, 2 }; in testNewArray()
750 assertEquals(cdim[counter], expr.getCreatedDimensions()); in testNewArray()