HomeSort by relevance Sort by last modified time
    Searched refs:kBatchMatMul (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
unroll_batch_matmul.cc 41 if (batch_op_it->get()->type != OperatorType::kBatchMatMul) {
propagate_fixed_sizes.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
model.h 38 kBatchMatMul,
754 BatchMatMulOperator() : Operator(OperatorType::kBatchMatMul) {}
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 41 constexpr char kBatchMatMul[] = "BatchMatMul";
166 {kBatchMatMul, wrap(&OpLevelCostEstimator::PredictBatchMatMul)},
617 if (op_features.op() != kBatchMatMul) {
    [all...]

Completed in 379 milliseconds