Searched refs:r_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | gru_ops.cc | 109 Tensor* r_tensor = nullptr; in Compute() local
112 &r_tensor)); in Compute()
155 r_tensor->matrix<T>(), u_tensor->matrix<T>(), c_tensor->matrix<T>(), in Compute()
195 const Tensor* r_tensor = nullptr; in Compute() local
196 OP_REQUIRES_OK(ctx, ctx->input("r", &r_tensor)); in Compute()
266 OP_REQUIRES(ctx, r_tensor->dim_size(0) == batch_size, in Compute()
268 "r.dims(0) != batch_size: ", r_tensor->dim_size(0), " vs. ", in Compute()
270 OP_REQUIRES(ctx, r_tensor->dim_size(1) == cell_size, in Compute()
272 "r.dims(1) != cell_size: ", r_tensor->dim_size(1), " vs. ", in Compute()
362 b_ru_tensor->vec<T>(), b_c_tensor->vec<T>(), r_tensor->matrix<T>(), in Compute()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 2556 ValueRef r_tensor = in compute_graph_round_trip_test() local
2558 ValueRef r_staging_in = graph.set_input_tensor(r_tensor); in compute_graph_round_trip_test()
2559 ValueRef r_staging_out = graph.set_output_tensor(r_tensor); in compute_graph_round_trip_test()
2564 vTensorPtr tensor = graph.get_tensor(r_tensor); in compute_graph_round_trip_test()
|