| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | parameter_value_benchmark.py | 36 def map_function(x): function
49 dataset = dataset.map(map_function)
89 dataset = dataset.map(map_function)
126 dataset = dataset.map(map_function)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 163 const std::function<float(float)>& map_function);
169 const std::function<float(float, float)>& map_function);
175 const std::function<float(float)>& map_function);
181 const std::function<float(float, float)>& map_function);
441 const std::function<float(float, int64_t, int64_t)>& map_function);
447 F&& map_function) { in MapArray4D() argument
450 return map_function(value); in MapArray4D()
460 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D() argument
468 map_function(input(plane, depth, height, width), plane, depth, in MapWithIndexArray4D()
482 F&& map_function) { in MapArray4D() argument
[all …]
|
| D | reference_util.cc | 639 const std::function<float(float)>& map_function) { in MapArray2D() argument
645 (*result)(i, j) = map_function(matrix(i, j)); in MapArray2D()
653 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
661 (*result)(i, j) = map_function(lhs(i, j), rhs(i, j)); in MapArray2D()
669 const std::function<float(float)>& map_function) { in MapArray3D() argument
677 (*result)(i, j, k) = map_function(array(i, j, k)); in MapArray3D()
686 const std::function<float(float, float)>& map_function) { in MapArray3D() argument
697 (*result)(i, j, k) = map_function(lhs(i, j, k), rhs(i, j, k)); in MapArray3D()
706 const std::function<float(float, int64_t, int64_t)>& map_function) { in MapWithIndexArray2D() argument
712 (*result)(i, j) = map_function(matrix(i, j), i, j); in MapWithIndexArray2D()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | grappler_test.py | 64 def map_function(x, y): function
72 dataset = dataset.map(map_function)
|
| D | make_deterministic_test.py | 330 def map_function(x): function
338 map_function, num_parallel_calls=2, deterministic=local_determinism)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | batch_test.py | 259 def map_function(x): function
267 map_function, num_parallel_calls=2, deterministic=local_determinism)
|
| D | map_test.py | 1280 def map_function(x): function
1289 map_function,
1304 def map_function(x, y): function
1309 dataset = apply_map(dataset, map_function)
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuScaleKernel.cpp | 241 static std::map<std::string, ScaleFunctionPtr> map_function = in configure() local
265 auto it = map_function.find(function_to_call); in configure()
266 if(it != map_function.end()) in configure()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1627 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations() argument
1629 called_computations_[i] = map_function(called_computations_[i]); in ReplaceCalledComputations()
|