| /external/tensorflow/tensorflow/python/profiler/ |
| D | tfprof_logger.py | 173 all_ops = {}
175 all_ops[entry.name] = entry
177 if op_name in all_ops:
178 all_ops[op_name].types.extend(entry.types)
179 if entry.float_ops > 0 and all_ops[op_name].float_ops == 0:
180 all_ops[op_name].float_ops = entry.float_ops
181 if entry.code_def.traces and not all_ops[op_name].code_def.traces:
182 all_ops[op_name].code_def.MergeFrom(entry.code_def)
184 all_ops[op_name] = entry
185 tmp_op_log.log_entries.extend(all_ops.values())
|
| /external/deqp-deps/SPIRV-Tools/test/reduce/ |
| D | operand_to_constant_test.cpp | 278 const auto all_ops = in TEST() local
281 ASSERT_TRUE(all_ops.size() > 100); in TEST()
301 ASSERT_EQ(all_ops.size(), ops_for_function_4.size() + in TEST()
|
| /external/swiftshader/third_party/SPIRV-Tools/test/reduce/ |
| D | operand_to_constant_test.cpp | 278 const auto all_ops = in TEST() local
281 ASSERT_TRUE(all_ops.size() > 100); in TEST()
301 ASSERT_EQ(all_ops.size(), ops_for_function_4.size() + in TEST()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_solve_op_test.py | 127 all_ops = []
139 all_ops += [s1, s2]
140 val = self.evaluate(all_ops)
141 for i in range(0, len(all_ops), 2):
|
| D | self_adjoint_eig_op_test.py | 57 all_ops = []
65 all_ops += [e1, v1, e2, v2]
69 all_ops += [e1, e2]
70 val = self.evaluate(all_ops)
|
| D | eig_op_test.py | 57 all_ops = []
65 all_ops += [e1, v1, e2, v2]
69 all_ops += [e1, e2]
70 val = self.evaluate(all_ops)
|
| D | matrix_inverse_op_test.py | 143 all_ops = []
149 all_ops += [inv1, inv2]
150 inv = self.evaluate(all_ops)
|
| D | matrix_square_root_op_test.py | 120 all_ops = [sqrt1, sqrt2]
121 sqrt = self.evaluate(all_ops)
|
| D | qr_op_test.py | 64 all_ops = []
76 all_ops += [q1, q2, r1, r2]
77 val = self.evaluate(all_ops)
|
| D | svd_op_test.py | 65 all_ops = []
78 all_ops += [s1, s2, u1, u2, v1, v2]
84 all_ops += [s1, s2]
85 val = self.evaluate(all_ops)
|
| D | control_flow_ops_py_test.py | 4041 all_ops = x.graph.get_operations()
4042 self.assertFalse(any(name in op.name for op in all_ops))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 470 all_ops = ops.get_collection("iterator_ops")
472 init_op, indices, values, dense_shape = all_ops
478 indices, values, dense_shape = all_ops[i:i + 3]
483 get_next_list.append(all_ops[i])
485 return all_ops[0], nest.pack_sequence_as(
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.cc | 61 std::vector<quant::QuantizeRegionOp> all_ops; in GetAllOps() local
62 all_ops.reserve(128); in GetAllOps()
63 func_.walk([&](quant::QuantizeRegionOp op) { all_ops.push_back(op); }); in GetAllOps()
64 return all_ops; in GetAllOps()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule_test.cc | 316 const HloVec all_ops({d00, d10, d11, d20, d21, d22, d30, d31, d40}); in TEST_F() local
323 for (const HloInstruction* op : all_ops) { in TEST_F()
329 for (const HloInstruction* op : all_ops) { in TEST_F()
355 for (const HloInstruction* op : all_ops) { in TEST_F()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | error_interpolation.py | 404 def traceback_files_common_prefix(all_ops): argument
417 for ops in all_ops:
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 1162 absl::flat_hash_set<string> all_ops(vall_ops.begin(), vall_ops.end()); in FindCompilationCandidates() local
1165 if (!all_ops.contains(string(s))) { in FindCompilationCandidates()
|
| D | mark_for_compilation_pass_test.cc | 1814 absl::flat_hash_set<string> all_ops(vall_ops.begin(), vall_ops.end()); in TEST() local
1820 ASSERT_TRUE(all_ops.contains(op)); in TEST()
|