Searched defs:map_function (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | grappler_test.py | 68 def map_function(x, y): function
|
| D | map_vectorization_test.py | 575 def map_function(x): 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()
483 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D()
|
| D | reference_util.cc | 637 const std::function<float(float)>& map_function) { in MapArray2D()
651 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
667 const std::function<float(float, int64, int64)>& map_function) { in MapWithIndexArray2D() argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 160 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function, in MakeLessStatefulFunction()
190 bool CanHoistRandomUniform(const FunctionDef& map_function, in CanHoistRandomUniform()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 1353 def map_function(x): function
1377 def map_function(x, y): function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1459 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations()
|