Searched defs:map_function (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | filter_fusion_test.py | 65 def testFilterFusion(self, map_function, predicates): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 443 F&& map_function) { in MapArray4D()
456 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D()
478 F&& map_function) { in MapArray4D()
491 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D()
|
| D | reference_util.cc | 702 const std::function<float(float)>& map_function) { in MapArray2D()
716 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
732 const std::function<float(float, int64, int64)>& map_function) { in MapWithIndexArray2D() argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 149 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function, in MakeLessStatefulFunction()
180 bool CanHoistRandomUniform(const FunctionDef& map_function, in CanHoistRandomUniform()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1211 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations()
|
| /external/v8/src/builtins/ |
| D | builtins-array-gen.cc | 2030 TNode<Object> map_function = args.GetOptionalArgumentValue(1); in TF_BUILTIN() local
|