Searched refs:result_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 197 std::vector<int64> result_dims, in BoundSamples() argument
225 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in BoundSamples()
229 xla::Broadcast(xla::Zero(ctx->builder(), warp_type), result_dims); in BoundSamples()
580 std::vector<int64> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
581 result_dims.push_back(data_channels); in Compile()
586 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in Compile()
590 xla::Broadcast(xla::Zero(ctx->builder(), data_type), result_dims); in Compile()
666 std::vector<int64> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
667 result_dims.push_back(2); in Compile()
671 BoundSamples(ctx, warp, warp_type, warp_shape, result_dims, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 874 DimensionVector result_dims; in CollapseFirstNDims() local
875 result_dims.push_back(prefix_dim); in CollapseFirstNDims()
877 std::back_inserter(result_dims)); in CollapseFirstNDims()
879 result_dims); in CollapseFirstNDims()
|