Home
last modified time | relevance | path

Searched refs:EachCell (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dprng_test.cc62 actual.EachCell<T>([=](absl::Span<const int64>, T value) { in UniformTest()
121 result.EachCell<bfloat16>([&](absl::Span<const int64>, bfloat16 value) { in XLA_TEST_F()
153 actual.EachCell<int32>( in UniformChiSquared()
Dreverse_test.cc94 expected.EachCell<float>([&](absl::Span<const int64> indices, float) { in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc202 root->literal().EachCell<NativeT>( in TEST_F()
Dhlo_evaluator_test.cc434 result.EachCell<NativeT>([&](absl::Span<const int64> indices, NativeT value) { in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h165 void EachCell(
978 void LiteralBase::EachCell( in EachCell() function
Dliteral_test.cc725 reshape.EachCell<float>([&](absl::Span<const int64> indices, float value) { in TEST_F()
Dliteral.cc718 result_literal.EachCell<NativeT>( in SliceInternal()