Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 208 for pushBack (0.14 sec)

  1. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

          std::vector<uint8_t>(expected_value.begin(), expected_value.end());
      input_model.buffers.push_back(std::move(model_metadata_buffer));
      auto metadata_t = std::make_unique<tflite::MetadataT>();
      metadata_t->name = tflite::optimize::kTfLiteReducedPrecisionKey;
      metadata_t->buffer = input_model.buffers.size() - 1;
      input_model.metadata.push_back(std::move(metadata_t));
    
      // Sparsify and create output model
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/schema/flatbuffer_compatibility_test.cc

                   const std::string &contents) {
      std::vector<const char *> include_directories;
      auto local_include_directory = flatbuffers::StripFileName(filename);
      include_directories.push_back(local_include_directory.c_str());
      include_directories.push_back(nullptr);
      if (!parser->Parse(contents.c_str(), include_directories.data(),
                         filename.c_str())) {
        fprintf(stderr, "Failed to parse flatbuffer schema '%s'\n",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 14:28:27 UTC 2024
    - 3.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc

              outputs_replaced.push_back(user.getResult());
              output_types.push_back(user.getType());
            } else if (isa<IntegerType>(result_element_type)) {
              // If the result is an integer tensor, then it doesn't require the
              // dequantize op in the pattern.
              outputs_replaced.push_back(result);
              output_types.push_back(result.getType());
            } else {
              return;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 06:04:36 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc

            values_for_results.push_back(output);
          }
        }
      }
      return values_for_results;
    }
    
    // Compute signature for raised func from arugments and outputs of
    // Operation partition.
    llvm::SmallVector<Type> TypesFromValues(
        const llvm::SmallVector<Value>& values) {
      llvm::SmallVector<Type> types;
      for (auto value : values) {
        types.push_back(value.getType());
      }
      return types;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc

        if (rhs_batch_idx.contains(i)) {
          if (rhs_only_batch) {
            rhs_eq.push_back(c);
            lhs_batch_dims.push_back(c);
            c++;
          } else {
            rhs_eq.push_back(lhs_batch_dims[batch_trace_idx]);
            batch_trace_idx++;
          }
        } else if (rhs_contract_idx.contains(i)) {
          absl::StrAppend(&rhs_eq,
                          std::string(1, lhs_contract_dims[contract_trace_idx]));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

              rmax = std::max(std::abs(rmin), std::abs(rmax));
              rmin = -rmax;
            }
            TensorRangeSanityCheck(op, rmin, rmax);
            mins.push_back(rmin);
            maxs.push_back(rmax);
          }
          quant_type = quantfork::fakeQuantAttrsToType(
              op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range,
              expressed, is_signed);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

          reshaped_shape_ints.push_back(outer_shape_ints[i]);
          reshaped_shape_ints.push_back(block_shape_ints[i]);
        }
        for (int64_t i = 1 + block_rank; i < input_rank; ++i) {
          reshaped_shape_vals.push_back(padded_shape_splits[i]);
          reshaped_shape_ints.push_back(padded_shape[i]);
        }
        auto reshaped_shape = ValuesToRank1(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfr/passes/decompose.cc

                  op->getLoc(), unconstrainted_tensor_type, op->getOperand(i));
              variadic_operands.push_back(casted);
            }
            auto build_list_op = builder.create<BuildListOp>(
                op->getLoc(), list_type, variadic_operands);
            new_operands.push_back(build_list_op.getOut());
          } else {
            auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc

      SmallVector<int64_t, 8> perm0_values;
      for (const auto &value : perm0.getValues<APInt>())
        perm0_values.push_back(value.getSExtValue());
    
      SmallVector<int64_t, 8> perm1_values;
      for (const auto &value : perm1.getValues<APInt>())
        perm1_values.push_back(value.getSExtValue());
    
      for (int i = 0; i < perm0_values.size(); ++i) {
        if (perm0_values[perm1_values[i]] != i) return false;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

        result_types.push_back(op->getOpResult(2).getType());
    
        // Populate attributes.
        SmallVector<NamedAttribute, 4> attributes;
        // Activation will always be tanh.
        attributes.push_back(rewriter.getNamedAttr("fused_activation_function",
                                                   rewriter.getStringAttr("TANH")));
        // cell_clip.
        attributes.push_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
Back to top