Searched refs:cdim (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dconcat_op.cc154 // Here, N=3, cdim=1, dims=4. The concatenated tensor has shape
169 const int64 cdim = concat_dim_literal.Get<int>({}); variable
171 VLOG(1) << "ConcatOffset " << cdim << "," << dims;
172 int32 axis = cdim < 0 ? cdim + dims : cdim;
/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op.cc287 // Here, N=3, cdim=1, dims=4. The concatenated tensor has shape
297 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); variable
299 int32 axis = cdim < 0 ? cdim + dims : cdim;
301 errors::InvalidArgument("Concat dim is out of range: ", cdim,
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dconcat_op_test.py614 cdim = constant_op.constant(1, dtypes.int32)
618 off = gen_array_ops._concat_offset(cdim, [s0, s1, s2])
624 cdim = constant_op.constant(1, dtypes.int32)
627 off = gen_array_ops._concat_offset(cdim, [s0, s1])
634 cdim = constant_op.constant(4, dtypes.int32)
637 off = gen_array_ops._concat_offset(cdim, [s0, s1])
644 cdim = constant_op.constant(1, dtypes.int32)
647 off = gen_array_ops._concat_offset(cdim, [s0, s1])
654 cdim = constant_op.constant(1, dtypes.int32)
657 off = gen_array_ops._concat_offset(cdim, [s
[all...]
/external/tremolo/Tremolo/
H A Dfloor1.c307 int cdim=info->klass[classv].class_dim; local
319 for(k=0;k<cdim;k++){
329 j+=cdim;
/external/tensorflow/tensorflow/core/util/sparse/
H A Dsparse_tensor.h446 const int cdim = (d < primary_dim) ? d : d + 1; local
447 CHECK_EQ(final_shape[cdim], st_shape[cdim])
450 << ", mismatched shape at dim: " << cdim
/external/tensorflow/tensorflow/compiler/tests/
H A Dconcat_ops_test.py300 cdim = constant_op.constant(1, dtypes.int32)
304 off = gen_array_ops._concat_offset(cdim, [s0, s1, s2])

Completed in 170 milliseconds