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 | 648 cdim = constant_op.constant(1, dtypes.int32)
652 off = gen_array_ops.concat_offset(cdim, [s0, s1, s2])
658 cdim = constant_op.constant(1, dtypes.int32)
661 off = gen_array_ops.concat_offset(cdim, [s0, s1])
668 cdim = constant_op.constant(4, dtypes.int32)
671 off = gen_array_ops.concat_offset(cdim, [s0, s1])
678 cdim = constant_op.constant(1, dtypes.int32)
681 off = gen_array_ops.concat_offset(cdim, [s0, s1])
689 cdim = constant_op.constant(1, dtypes.int32)
692 off = gen_array_ops.concat_offset(cdim, [s0, s1])
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op.cc | 315 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); in Compute() local
317 int32 axis = cdim < 0 ? cdim + dims : cdim; in Compute()
319 errors::InvalidArgument("Concat dim is out of range: ", cdim, in Compute()
|
| /external/tremolo/Tremolo/ |
| D | floor1.c | 307 int cdim=info->klass[classv].class_dim; in floor1_inverse1() local
319 for(k=0;k<cdim;k++){ in floor1_inverse1()
329 j+=cdim; in floor1_inverse1()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 523 const int cdim = (d < primary_dim) ? d : d + 1; in Concat() local
524 DCHECK_EQ(final_shape[cdim], st_shape[cdim]) in Concat()
527 << ", 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])
|