Searched refs:broadcasted_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 198 std::vector<int64> broadcasted_dims, int64 last_warp_dim, in BoundSamples() argument
225 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in BoundSamples()
282 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in CalculateGradData() local
283 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in CalculateGradData()
285 reshaped_result_dims, broadcasted_dims, in CalculateGradData()
583 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local
584 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in Compile()
586 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in Compile()
668 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local
669 std::iota(broadcasted_dims.begin(), broadcasted_dims.end(), 0); in Compile()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 527 broadcasted_dims = reversed(list(six.moves.zip_longest(
534 for (dim_x, dim_y) in broadcasted_dims:
|