Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for EndMap (0.17 sec)

  1. tensorflow/compiler/mlir/lite/utils/tftext_utils.cc

      std::string reduction_type_str(reduction_type.getValue().data(),
                                     reduction_type.getValue().size());
      fbb.String("reduction_type", reduction_type_str);
    
      fbb.EndMap(start_map);
      fbb.Finish();
      custom_option_buffer.assign(fbb.GetBuffer().begin(), fbb.GetBuffer().end());
      return success();
    }
    
    LogicalResult ConvertNgrams(func::FuncOp func, llvm::StringRef api,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc

          for (const NamedAttribute& pair : composite.getCompositeAttributes()) {
            // Allows skipping unsupported attributes, will warn.
            (void)BuildOption(fbb.get(), op, pair);
          }
          fbb->EndMap(map_start);
          fbb->Finish();
          custom_option_buffer.assign(fbb->GetBuffer().begin(),
                                      fbb->GetBuffer().end());
    
          // Build TFL custom op, replace composite with custom op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

      if (!use_regular_nms) {
        return func.emitError()
               << "use_regular_nms attribute is not set or not a bool";
      }
      fbb.Int("use_regular_nms", use_regular_nms.getValue());
    
      fbb.EndMap(start_map);
      fbb.Finish();
      custom_option_buffer.assign(fbb.GetBuffer().begin(), fbb.GetBuffer().end());
      return success();
    }
    
    LogicalResult ConvertSSDPostProcessFunc::AddIntAttr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc

          fbb.String(attr.first.data(), string_attr.getValue().str());
        } else {
          // TODO(b/201482289): support other data types.
          return failure();
        }
      }
    
      fbb.EndMap(start_map);
      fbb.Finish();
      custom_option_buffer.assign(fbb.GetBuffer().begin(), fbb.GetBuffer().end());
      return success();
    }
    
    // Convert func annotated with `tfl_fusable_op` attribute to tfl custom op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

            }
            break;
          default:
            emitWarning(loc, "ignoring unsupported attribute type with key: ")
                << key;
            break;
        }
      }
      flex_builder->EndMap(map_start);
      flex_builder->Finish();
      return flex_builder;
    }
    
    uint32_t Translator::GetOpcodeIndex(const std::string& op_name,
                                        tflite::BuiltinOperator builtin) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top