Searched defs:map_function (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | reference_util.h | 447 F&& map_function) { in MapArray4D()
460 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D()
482 F&& map_function) { in MapArray4D()
496 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D()
|
| H A D | reference_util.cc | 639 const std::function<float(float)>& map_function) { in MapArray2D()
653 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
669 const std::function<float(float)>& map_function) { in MapArray3D()
686 const std::function<float(float, float)>& map_function) { in MapArray3D() argument
706 const std::function<float(float, int64_t, int64_t)>& map_function) { in MapWithIndexArray2D() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| H A D | grappler_test.py | 64 def map_function(x, y): function
|
| H A D | make_deterministic_test.py | 330 def map_function(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| H A D | parameter_value_benchmark.py | 36 def map_function(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | batch_test.py | 259 def map_function(x): function
|
| H A D | map_test.py | 1280 def map_function(x): function
1304 def map_function(x, y): function
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuScaleKernel.cpp | 241 static std::map<std::string, ScaleFunctionPtr> map_function = in configure() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.h | 1627 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations()
|