Home
last modified time | relevance | path

Searched refs:rhs_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc63 se::DeviceMemory<Element> rhs_data(rhs_matrix.data); in DoGemm() local
77 lhs_data, /*leading dim of LHS=*/lhs_matrix.num_rows, rhs_data, in DoGemm()
91 /*leading dim of LHS=*/lhs_matrix.num_rows, lhs_stride, rhs_data, in DoGemm()
128 se::DeviceMemory<Element> rhs_data(rhs_matrix.data); in DoGemmWithAlgorithm() local
142 /*leading dim of LHS=*/lhs_matrix.num_rows, rhs_data, in DoGemmWithAlgorithm()
318 se::DeviceMemoryBase rhs_data = in ExecuteOnStream() local
390 rhs_data, rhs_shape_, dim_nums.rhs_contracting_dimensions(0) == col_dim); in ExecuteOnStream()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc751 const auto& rhs_data = rhs_array.GetBuffer<A>().data; in CompareArrayBuffers() local
752 CHECK_EQ(lhs_data.size(), rhs_data.size()) in CompareArrayBuffers()
755 if (lhs_data[i] != rhs_data[i]) { in CompareArrayBuffers()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc244 std::unique_ptr<GlobalData> rhs_data = in XLA_TEST_F() local
254 ComputeAndCompareR1<uint64>(&b, expected, {lhs_data.get(), rhs_data.get()}); in XLA_TEST_F()
283 std::unique_ptr<GlobalData> rhs_data = in XLA_TEST_F() local
293 ComputeAndCompareR1<int64>(&b, expected, {lhs_data.get(), rhs_data.get()}); in XLA_TEST_F()
/external/epid-sdk/epid/common/math/unittests/
Dffelement-test.cc82 auto rhs_data = rhs.data(); in operator ==() local
83 return lhs_data.size() == rhs_data.size() && in operator ==()
84 std::equal(lhs_data.begin(), lhs_data.end(), rhs_data.begin()); in operator ==()
/external/v8/src/builtins/
Dbuiltins-string-gen.cc256 Node* rhs_data = DirectStringData(rhs, rhs_instance_type); in StringEqual_Loop() local
274 Load(rhs_type, rhs_data, in StringEqual_Loop()