Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for out_specs (0.13 sec)

  1. tensorflow/compiler/mlir/tfrt/tests/saved_model/saved_model_test.cc

      ASSERT_EQ(in_specs.size(), 1);
      ASSERT_EQ(in_specs[0].first, tensorflow::DT_INT32);
      ASSERT_TRUE(in_specs[0].second.IsIdenticalTo(PartialTensorShape({1, 3})));
    
      ASSERT_EQ(out_specs.size(), 1);
      ASSERT_EQ(out_specs[0].first, tensorflow::DT_INT32);
      ASSERT_TRUE(out_specs[0].second.IsIdenticalTo(PartialTensorShape({1, 1})));
    
      ASSERT_EQ(bound_inputs.size(), 2);
    
      auto global_tensor =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 13 01:17:29 UTC 2023
    - 9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/device_target.cc

      auto rop = llvm::dyn_cast<quantfork::QuantizeRegionOp>(op);
      if (!rop) return failure();
    
      llvm::SmallVector<Type, 4> input_specs, out_specs;
      for (auto spec : rop.getInputSpecs()) {
        input_specs.push_back(spec.cast<TypeAttr>().getValue());
      }
      for (auto spec : rop.getOutputSpecs()) {
        out_specs.push_back(spec.cast<TypeAttr>().getValue());
      }
    
      auto in_spec = input_specs[0].dyn_cast<UniformQuantizedType>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 08 10:41:08 UTC 2024
    - 7.3K bytes
    - Viewed (0)
Back to top