Searched refs:operation_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_precision_test.cc | 462 int operation_index);
466 int operation_index = GetParam(); in XLA_TEST_P() local
467 DoIt<Eigen::half, uint16_t>(f16_exponent_sizes[operation_index], in XLA_TEST_P()
468 f16_mantissa_sizes[operation_index], in XLA_TEST_P()
469 f16_test_values, operation_index); in XLA_TEST_P()
473 int operation_index = GetParam(); in XLA_TEST_P() local
474 DoIt<bfloat16, uint16_t>(bf16_exponent_sizes[operation_index], in XLA_TEST_P()
475 bf16_mantissa_sizes[operation_index], in XLA_TEST_P()
476 bf16_test_values, operation_index); in XLA_TEST_P()
480 int operation_index = GetParam(); in XLA_TEST_P() local
[all …]
|
| /external/vulkan-validation-layers/layers/ |
| D | gpu_validation.cpp | 817 const uint32_t operation_index, std::string &msg) { in GenerateCommonMessage() argument
839 strm << "Index " << operation_index << ". " in GenerateCommonMessage()
1067 … uint32_t operation_index, uint32_t *const debug_output_buffer) { in AnalyzeAndReportError() argument
1106 operation_index, common_message); in AnalyzeAndReportError()
1131 uint32_t operation_index = 0; in ProcessInstrumentationBuffer() local
1133 operation_index = draw_index; in ProcessInstrumentationBuffer()
1135 operation_index = compute_index; in ProcessInstrumentationBuffer()
1137 operation_index = ray_trace_index; in ProcessInstrumentationBuffer()
1142 …alyzeAndReportError(cb_node, queue, buffer_info.pipeline_bind_point, operation_index, (uint32_t *)… in ProcessInstrumentationBuffer()
|
| D | core_validation.h | 1243 …CMD_BUFFER_STATE* cb_node, VkQueue queue, VkPipelineBindPoint bind_point, uint32_t operation_index,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1522 const int operation_index = item.index(); in BuildSubGraph() local
1608 operation_index_to_operator_index.try_emplace(operation_index, in BuildSubGraph()
|