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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
resampler_ops.cc 198 std::vector<int64> broadcasted_dims, int64 last_warp_dim,
225 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims);
282 std::vector<int64> broadcasted_dims(warp_dims.size() - 1);
283 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0);
285 reshaped_result_dims, broadcasted_dims,
583 std::vector<int64> broadcasted_dims(warp_dims.size() - 1);
584 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0);
586 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims);
    [all...]
  /external/tensorflow/tensorflow/python/framework/
common_shapes.py 527 broadcasted_dims = reversed(list(six.moves.zip_longest(
534 for (dim_x, dim_y) in broadcasted_dims:

Completed in 1593 milliseconds