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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
resampler_ops.cc 401 xla::DotDimensionNumbers dot_dims; local
403 dot_dims.add_lhs_batch_dimensions(i);
404 dot_dims.add_rhs_batch_dimensions(i);
406 dot_dims.add_lhs_contracting_dimensions(warp_shape.dims() - 1);
407 dot_dims.add_rhs_contracting_dimensions(warp_shape.dims() - 1);
415 neighbors_data, dot_dims, /*precision_config=*/nullptr);
423 neighbors_data, dot_dims, /*precision_config=*/nullptr);
431 neighbors_data, dot_dims, /*precision_config=*/nullptr);
439 neighbors_data, dot_dims, /*precision_config=*/nullptr);
538 xla::DotDimensionNumbers dot_dims; variable
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc 265 std::vector<int64> dot_dims = batch_dim_sizes; local
266 dot_dims.push_back(lhs_non_contracting_size);
267 dot_dims.push_back(rhs_non_contracting_size);
278 ShapeUtil::MakeShape(original_dot->shape().element_type(), dot_dims),

Completed in 475 milliseconds