Searched refs:a4drhs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 86 Array4D<float> a4drhs(rhs.n1(), rhs.n2(), rhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated() local
87 a4drhs.Each([&](absl::Span<const int64> indices, float* value_ptr) { in ConvArray3DGeneralDimensionsDilated()
97 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
|