HomeSort by relevance Sort by last modified time
    Searched refs:inner_dim_size (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
segment_reduction_ops_gpu.cu.cc 52 const Index inner_dim_size,
58 const Index segment_offset = stripe_index % inner_dim_size;
60 stripe_index / inner_dim_size * Index(OuterDimTileSize);
78 last_output_segment_id * inner_dim_size + segment_offset;
88 sum += ldg(input + (input_outer_dim_index_base + j) * inner_dim_size +
96 last_output_segment_id * inner_dim_size + segment_offset;
103 // 'segment_ids' mapping and 'inner_dim_size'.
106 const Index inner_dim_size,
110 const Index input_total_size = input_outer_dim_size * inner_dim_size;
111 const Index output_total_size = output_outer_dim_size * inner_dim_size;
    [all...]
ops_util_test.cc 362 // For multidimensional tensors, alignment is dictated by inner_dim_size.
363 int64 inner_dim_size = EIGEN_MAX_ALIGN_BYTES;
364 Tensor t(DT_FLOAT, TensorShape({3, inner_dim_size}));
371 CHECK(sliced.CopyFrom(t.Slice(start, end), TensorShape({1, inner_dim_size})));
378 // For multidimensional tensors, alignment is dictated by inner_dim_size.
379 int64 inner_dim_size = EIGEN_MAX_ALIGN_BYTES + 1; local
380 Tensor t(DT_FLOAT, TensorShape({3, inner_dim_size}));
387 CHECK(sliced.CopyFrom(t.Slice(start, end), TensorShape({1, inner_dim_size})));

Completed in 72 milliseconds