Home
last modified time | relevance | path

Searched refs:kBatch (Results 1 – 6 of 6) sorted by relevance

/external/skia/bench/
DBulkRectBench.cpp27 kBatch, // Bulk API submission, one call to draw every rectangle enumerator
52 if (kDrawMode == DrawMode::kBatch && kImageMode == ImageMode::kNone) { in isSuitableFor()
81 if (kDrawMode == DrawMode::kBatch) { in computeName()
92 SkASSERT(kDrawMode == DrawMode::kBatch); in drawImagesBatch()
131 SkASSERT(kDrawMode == DrawMode::kBatch); in drawSolidColorsBatch()
242 if (kDrawMode == DrawMode::kBatch) { in onDraw()
248 if (kDrawMode == DrawMode::kBatch) { in onDraw()
269 ADD_BENCH(n, layout, ImageMode::kShared, DrawMode::kBatch) \
271 ADD_BENCH(n, layout, ImageMode::kUnique, DrawMode::kBatch) \
273 ADD_BENCH(n, layout, ImageMode::kNone, DrawMode::kBatch) \
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc347 constexpr int kBatch = 2; in TEST() local
351 static float vector[kCol * kBatch] = {1.0, -1.0, 1.0, -1.0, // in TEST()
353 std::vector<float> output(kRow * kBatch); in TEST()
355 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST()
1483 const int kBatch = 2; in TEST() local
1530 float vector[kBatch * kCol] = { in TEST()
1544 std::vector<float> dense_output(kRow * kBatch, 0.0); in TEST()
1545 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST()
1553 std::vector<float> sparse_output(kRow * kBatch, 0.0); in TEST()
1555 matrix_values, ledger, kRow, kCol, vector, kBatch, sparse_output.data()); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc515 const int32 kBatch = operand_shape[1]; in RunR3Contiguous() local
517 Array3D<T> input_values(kSeq, kBatch, kDim); in RunR3Contiguous()
518 Array3D<T> update_values(size, kBatch, kDim); in RunR3Contiguous()
519 Array3D<T> expected_values(kSeq, kBatch, kDim); in RunR3Contiguous()
530 for (int j = 0; j < kBatch; j++) { in RunR3Contiguous()
/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h76 kBatch = 1, enumerator
94 return kBatch; in GetDimensionType()
501 if (label_types[label] == kBroadcasting || label_types[label] == kBatch) { in ReduceOperand()
520 const int64 output_size = reshape[kBroadcasting] * reshape[kBatch] * in ReduceOperand()
657 if (label_types[label] == EinsumHelper::kBatch) in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc4822 constexpr int kBatch = 8; in TEST_P() local
4826 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P()
4828 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P()
4864 constexpr int kBatch = 8; in TEST_P() local
4868 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P()
4871 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P()
4910 constexpr int kBatch = 8; in TEST_P() local
4914 auto lhs_shape = ShapeUtil::MakeShape(F32, {kBatch, kFeature}); in TEST_P()
4916 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P()
4975 constexpr int kBatch = 8; in TEST_P() local
[all …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc1059 void VerifyFusedMatMul(const int kBatch, const int kInputChannel, in VerifyFusedMatMul() argument
1132 CommonTestUtilities<T>::VerifyFusedMatrixClose(kInputChannel, kBatch, in VerifyFusedMatMul()