Home
last modified time | relevance | path

Searched refs:outer_dims (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc25 std::vector<int64> outer_dims; in Run() local
31 outer_dims.push_back(dimension); in Run()
48 static_cast<double>(target_partition_count), 1.0 / outer_dims.size()); in Run()
52 std::vector<int64> dimension_partition_counts(outer_dims.size()); in Run()
53 for (int64 i = 0; i < outer_dims.size(); ++i) { in Run()
55 std::min(static_cast<int64>(shape_.dimensions(outer_dims[i])), in Run()
79 shape_.dimensions(outer_dims[i]) - dimension_partition_counts[i], in Run()
Dir_emitter.cc1993 std::vector<int64> outer_dims; in HandleSlice() local
1995 outer_dims.push_back(LayoutUtil::Major(layout, i)); in HandleSlice()
2001 outer_dims.push_back(memcpy_dim); in HandleSlice()
2006 const int64 num_outer_loops = outer_dims.size(); in HandleSlice()
2009 loops.AddLoopsForShapeOnDimensions(slice->shape(), outer_dims, "slice"); in HandleSlice()
2411 std::vector<int64> outer_dims(std::next(concat_dim_layout_itr), in EmitFastConcatenate() local
2421 loops.AddLoopsForShapeOnDimensions(output_shape, outer_dims, "concat"); in EmitFastConcatenate()
2428 if (!outer_dims.empty()) { in EmitFastConcatenate()
2471 if (!outer_dims.empty()) { in EmitFastConcatenate()
/external/tensorflow/tensorflow/core/kernels/
Dscatter_nd_op.cc100 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local
102 for (int i = 0; i < outer_dims; ++i) { in Compute()
111 const int64 ix = indices.shape().dim_size(outer_dims); in Compute()
113 c, updates.shape().dims() - outer_dims == shape.dims() - ix, in Compute()
118 for (int i = 0; i + outer_dims < updates.shape().dims(); ++i) { in Compute()
120 c, updates.shape().dim_size(i + outer_dims) == shape.dim_size(ix + i), in Compute()
124 " must match the inner ", updates.shape().dims() - outer_dims, in Compute()
171 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local
173 for (int i = 0; i < outer_dims; ++i) { in Compute()
182 const int64 ix = indices.shape().dim_size(outer_dims); in Compute()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py445 outer_dims = array_ops.size(outer_shape)
449 outer_axes_indices = math_ops.range(outer_dims)
450 inner_axes_indices = math_ops.range(outer_dims + 1,
451 outer_dims + 1 + inner_dims)
464 [[outer_dims], outer_axes_indices, inner_axes_indices], 0)
/external/tensorflow/tensorflow/python/kernel_tests/
Dgather_op_test.py139 outer_dims = axis
144 dest_slice = ((slice(None),) * outer_dims + (dest_index,) +
146 source_slice = ((slice(None),) * outer_dims + (source_index,) +
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py97 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
98 outer_paddings = _array_ops.zeros([outer_dims], fft_length.dtype)
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2885 const int64 outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
2895 c->Subshape(indices_shape, 0, outer_dims, &prefix_indices)); in ScatterNdShapeHelper()
2898 c->Subshape(updates_shape, 0, outer_dims, &prefix_updates)); in ScatterNdShapeHelper()
2903 "The outer ", outer_dims, in ScatterNdShapeHelper()
2905 " must match the outer ", outer_dims, in ScatterNdShapeHelper()
2914 c->Subshape(updates_shape, outer_dims, &suffix_updates)); in ScatterNdShapeHelper()
2920 " must match the inner ", c->Rank(updates_shape) - outer_dims, in ScatterNdShapeHelper()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1755 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local
1758 outer_dims + dot->dot_dimension_numbers().lhs_batch_dimensions_size()); in HandleDot()