Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 299 for pushBack (0.24 sec)

  1. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/importer_test_min_max.cc

            input_tensor->quantization->scale.clear();
            input_tensor->quantization->zero_point.clear();
            input_tensor->quantization->min.push_back(-1.0);
            input_tensor->quantization->max.push_back(1.0);
    
            auto& output_tensor = sub_graph->tensors[op->outputs[0]];
            auto shape = output_tensor->shape;
            output_tensor->quantization->scale.clear();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc

      std::vector<TFE_TensorHandle*> compute_fn_inputs;
      TFE_TensorHandle* input_a = TestScalarTensorHandle(ctx, 2.0f);
      TFE_TensorHandle* input_b = TestScalarTensorHandle(ctx, 1.0f);
      compute_fn_inputs.push_back(input_a);
      compute_fn_inputs.push_back(input_b);
    
      TFE_Op* compute_fn_op = TF_ConcreteFunctionMakeCallOp(
          compute_fn, compute_fn_inputs.data(), compute_fn_inputs.size(), status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 08:08:45 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

            launch_results.push_back(result);
            launch_result_types.push_back(result.getType());
          }
        }
      }
    
      before ? builder->setInsertionPoint(op) : builder->setInsertionPointAfter(op);
      auto launch = builder->create<mlir::tf_device::LaunchOp>(
          op->getLoc(), builder->getStringAttr(host_device), launch_result_types);
      launch.getBody().push_back(launch_block);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

        // 1.0e-6.
        if (max - min > kNearZeroTolerance) {
          effective_mins.push_back(min);
          effective_maxs.push_back(max);
        } else {
          effective_mins.push_back(std::min(min, -kSmallestHalfRange));
          effective_maxs.push_back(std::max(max, kSmallestHalfRange));
        }
      }
    }
    
    // Sets the min / max, scale and zero_points from the fake quant num_bits
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc

            return rewriter.notifyMatchFailure(
                second, "non-concat dims have mismatching slice bounds");
        }
    
        new_start.push_back(first_start);
        new_limit.push_back(second_limit);
        new_slice_shape.push_back(second_limit - first_start);
      }
    
      auto new_slice = rewriter.create<mhlo::SliceOp>(
          FusedLoc::get(first->getContext(), {first.getLoc(), second.getLoc()}),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 26.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/cc/pre_calibration_test.cc

      EXPECT_THAT(pre_calibration_result, IsOk());
    
      SmallVector<func::FuncOp> func_ops;
      for (auto func_op : pre_calibration_result->getOps<func::FuncOp>()) {
        func_ops.push_back(func_op);
      }
      ASSERT_THAT(func_ops, SizeIs(2));
      EXPECT_THAT(func_ops, Contains(HasSymName("main")));
      EXPECT_THAT(func_ops, Contains(HasSymName("composite_dot_general_fn_1")));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 21:41:08 UTC 2024
    - 6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

            auto dequantize_op = builder->create<TFL::DequantizeOp>(
                op->getLoc(), dequantized_input_type, input.get());
            dequantized_inputs.push_back(dequantize_op);
          } else {
            dequantized_inputs.push_back(input.get());
          }
        }
    
        // Result types.
        SmallVector<Type, 4> result_types;
        for (auto result_type : op->getResultTypes()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

            int64_t size =
                padding_low[i] + shape[i] + (shape[i] - 1) * padding_interm[i];
            padding_high.push_back(input_dim - size);
          } else {
            dims_to_reverse.push_back(i);
            padding_high.push_back(input_dim - begin_indices[i] - 1);
            padding_interm.push_back(-strides[i] - 1);
    
            // Pad the lower dimension up to the expected input shape.
            int64_t size =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      for (int input = 0, e = num_operands + 1; input != e; ++input) {
        if (input < dst_input) {
          state.operands.push_back(dst->getOperand(input));
        } else if (input == dst_input) {
          state.operands.push_back(source->getResult(0));
        } else {
          state.operands.push_back(dst->getOperand(input - 1));
        }
      }
      state.attributes.assign(dst->getAttrs().begin(), dst->getAttrs().end());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      min_maxs.reserve(mins.size() * 2);
      for (int i = 0, end = mins.size(); i < end; ++i) {
        llvm::APFloat min(mins[i]);
        llvm::APFloat max(maxs[i]);
        min_maxs.push_back(min);
        min_maxs.push_back(max);
      }
      // The layer stats contain only the first min/max pairs.
      mlir::ElementsAttr layer_stats = mlir::DenseFPElementsAttr::get(
          tensorflow::GetTypeFromTFTensorShape({2}, b.getF32Type()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top