Home
last modified time | relevance | path

Searched defs:o_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/
H A Dlstm_ops.cc384 Tensor* o_tensor = nullptr; in Compute() local
557 const Tensor* o_tensor = nullptr; in Compute() local
1044 Tensor o_tensor = slicer.OutputSlice(o_out, t, "o_out"); in Compute() local
1331 const Tensor& o_tensor = slicer.InputSlice(*o_out, t, "o_out"); in Compute() local
/aosp_15_r20/external/pytorch/test/inductor/
H A Dcustom_ops.cpp34 const std::optional<Tensor>& o_tensor, in fn_with_all_inputs_impl()
244 const std::optional<Tensor>& o_tensor, in fn_with_all_inputs_meta()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantization_utils_test.cc49 Tensor o_tensor(DT_QUINT8, TensorShape{values_count}); in TestRequantizeMany() local
91 Tensor o_tensor(DT_QINT32, TensorShape{values_count}); in TestRequantizeMany8To32Bit() local