Searched refs:copy_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 71 def copy_tensor(dtype=dtypes.int32): function
80 copy_tensor()
88 copy_tensor(dtypes.float32)
89 copy_tensor()
95 copy_tensor()
103 copy_tensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 224 "copy_tensor.h",
291 ":copy_tensor",
363 name = "copy_tensor",
364 srcs = ["copy_tensor.cc"],
365 hdrs = ["copy_tensor.h"],
397 ":copy_tensor",
1134 ":copy_tensor",
1307 ":copy_tensor",
1324 ":copy_tensor",
1343 ":copy_tensor",
[all …]
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.c | 75 static void copy_tensor(const TENSOR *src, int copy_channels, int dst_offset, in copy_tensor() function
130 copy_tensor(dst, dst->channels, 0, &t); in concat_tensor()
138 copy_tensor(src, src->channels, dst_channels, dst); in concat_tensor()
341 copy_tensor(layer_active_tensor, copy_channels, 0, &branch_output[b]); in copy_active_tensor_to_branches()
979 copy_tensor(&tensor2[b], tensor2[b].channels, num_chs, in av1_cnn_predict_c()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 186 "//tensorflow/core/common_runtime:copy_tensor",
|