Searched defs:map_function (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | grappler_test.py | 68 def map_function(x, y): function
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 435 F&& map_function) { in MapArray4D()
448 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D()
470 F&& map_function) { in MapArray4D()
484 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D()
|
| D | reference_util.cc | 638 const std::function<float(float)>& map_function) { in MapArray2D()
652 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
668 const std::function<float(float, int64_t, int64_t)>& map_function) { in MapWithIndexArray2D() argument
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | parameter_value_benchmark.py | 40 def map_function(x): function
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | batch_test.py | 262 def map_function(x): function
|
| D | map_test.py | 1282 def map_function(x): function
1306 def map_function(x, y): function
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NESelectKernel.cpp | 186 static std::map<std::string, SelectFunction *> map_function; in configure() local
|
| D | NEFuseBatchNormalizationKernel.cpp | 470 static std::map<std::string, FuseBatchNormFunction *> map_function = in configure() local
|
| D | NEElementwiseOperationKernel.cpp | 1112 … std::map<std::string, NEElementwiseOperationKernel::ElementwiseFunction *> map_function) in configure_func()
1136 static std::map<std::string, NEElementwiseOperationKernel::ElementwiseFunction *> map_function = in configure_arithm_func() local
1155 static std::map<std::string, NEElementwiseOperationKernel::ElementwiseFunction *> map_function = in configure_comp_func() local
|
| D | NEScaleKernel.cpp | 156 static std::map<std::string, ScaleFunctionPtr> map_function = in configure() local
|
| D | NEArithmeticAdditionKernel.cpp | 933 static std::map<std::string, AddFunction *> map_function = in configure() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1564 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations()
|