Searched refs:dims64 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 668 std::vector<int64_t> dims64 = in ScopedTensorDescriptor() local
676 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in ScopedTensorDescriptor()
764 std::vector<int64_t> dims64 = in ScopedFilterDescriptor() local
772 absl::c_transform(dims64, std::back_inserter(dims), in ScopedFilterDescriptor()
4236 std::vector<int64_t> dims64 = in DoPoolBackward() local
4244 for (auto& x : dims64) dest2_size *= x; in DoPoolBackward()
4371 std::vector<int64_t> dims64 = in DoNormalizeBackwardWithDimensions() local
4377 std::transform(dims64.cbegin(), dims64.cend(), dimsint.begin(), in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 624 std::vector<int64_t> dims64 = in CudnnTensorDescriptor() local
632 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in CudnnTensorDescriptor()
6186 std::vector<int64_t> dims64 = in GetTensorSplits() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/ |
| D | large.disabled.mlir | 95 …%dims64 = "tf.Const"() { value = dense<[0, 2]> : tensor<2xi32>, device = "/job:localhost/replica:0…
96 …%64 = "tf.Sum"(%V__2, %dims64) { keep_dims = false, device = "/job:localhost/replica:0/task:0/devi…
|