Searched refs:cdim (Results 1 – 6 of 6) 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/python/kernel_tests/ |
| D | concat_op_test.py | 649 cdim = constant_op.constant(1, dtypes.int32)
653 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
659 cdim = constant_op.constant(1, dtypes.int32)
662 off = gen_array_ops.concat_offset(cdim, [s0, s1])
669 cdim = constant_op.constant(4, dtypes.int32)
672 off = gen_array_ops.concat_offset(cdim, [s0, s1])
679 cdim = constant_op.constant(1, dtypes.int32)
682 off = gen_array_ops.concat_offset(cdim, [s0, s1])
690 cdim = constant_op.constant(1, dtypes.int32)
693 off = gen_array_ops.concat_offset(cdim, [s0, s1])
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op.cc | 318 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); in Compute() local
320 int32 axis = cdim < 0 ? cdim + dims : cdim; in Compute()
322 errors::InvalidArgument("Concat dim is out of range: ", cdim, in Compute()
|
| /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 | 409 const int cdim = (d < primary_dim) ? d : d + 1; in Concat() local
410 DCHECK_EQ(final_shape[cdim], st_shape[cdim]) in Concat()
413 << ", mismatched shape at dim: " << cdim in Concat()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | concat_ops_test.py | 338 cdim = constant_op.constant(1, dtypes.int32)
342 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
|