Searched refs:cdim (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | concat_op.cc | 161 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/ |
D | concat_op.cc | 284 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/ |
D | concat_op_test.py | 676 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/ |
D | floor1.c | 308 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/ |
D | sparse_tensor.h | 444 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/ |
D | sharding_propagation.cc | 1164 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/ |
D | concat_ops_test.py | 340 cdim = constant_op.constant(1, dtypes.int32) 344 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
|
/external/javassist/src/test/javassist/ |
D | JvstTest2.java | 739 int cdim[] = { 1, 1, 2, 1, 1, 2, 2 }; in testNewArray() 750 assertEquals(cdim[counter], expr.getCreatedDimensions()); in testNewArray()
|