Searched refs:cdim (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | concat_op.cc | 154 // 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 D | concat_op.cc | 287 // 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 D | concat_op_test.py | 614 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 D | floor1.c | 307 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 D | sparse_tensor.h | 446 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 D | concat_ops_test.py | 300 cdim = constant_op.constant(1, dtypes.int32) 304 off = gen_array_ops._concat_offset(cdim, [s0, s1, s2])
|
Completed in 170 milliseconds