Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 76 for output_types (0.41 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/resource-alias-analysis-test.mlir

      %iter_handle1 = "tf.IteratorV2"() {container = "c", shared_name = "v0", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res
      // expected-remark@below {{Result #0, ID 1 : 0, 1}}
      %iter_handle2 = "tf.IteratorV2"() {container = "c", shared_name = "v1", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res
      func.return
    }
    
    // -----
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jul 21 17:19:47 UTC 2023
    - 22.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc

          }
    
          // Collect all the quantized outputs and replace them by the results of
          // the new quantized op.
          llvm::SmallDenseMap<Value, int> outputs_replaced;
          SmallVector<Type, 4> output_types;
          output_types.reserve(quantizing_op->getNumResults());
          for (const auto& enumerated_result :
               llvm::enumerate(quantizing_op->getResults())) {
            Value result = enumerated_result.value();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 23.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc

        arg_types.push_back(arg.getType());
        arg_locs.push_back(arg.getLoc());
      }
    
      SmallVector<Type> output_types;
      for (const Value output : xla_call_module_op.getOutput()) {
        output_types.push_back(output.getType());
      }
    
      entry_func_op.setFunctionType(
          rewriter.getFunctionType(arg_types, output_types));
    
      // Replace argument types and locs.
      Block& entry = entry_func_op->getRegion(0).front();
    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/cc/framework/cc_op_gen.cc

                           ";\n");
      }
      if (op_info.output_types.empty()) {
        // Allow casting this class to Operation.
        strings::StrAppend(&class_decl,
                           "  operator ::tensorflow::Operation() const { "
                           "return operation; }\n");
      } else if (op_info.output_types.size() == 1) {
        if (op_info.is_list_output[0]) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 17:22:47 UTC 2023
    - 17K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

        llvm::SmallVector<Type, 4> output_types;
        auto result = tensorflow::GetOutputTypesForLogicalDeviceComputation(
            core, output_sharding_config, cluster_func, &output_types,
            &(*cluster_to_core_index)[core]);
        if (failed(result)) return failure();
    
        for (Type t : output_types) concatenated_output_types.emplace_back(t);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

          }
    
          // Collect all the quantized outputs and replace them by the results of
          // the new quantized op.
          llvm::SmallDenseMap<Value, int> outputs_replaced;
          SmallVector<Type, 4> output_types;
          output_types.reserve(quantizing_op->getNumResults());
          for (const auto& enumerated_result :
               llvm::enumerate(quantizing_op->getResults())) {
            Value result = enumerated_result.value();
    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/jit/xla_cluster_util.cc

      return std::find(node.input_types().begin(), node.input_types().end(),
                       DT_RESOURCE) != node.input_types().end() ||
             std::find(node.output_types().begin(), node.output_types().end(),
                       DT_RESOURCE) != node.output_types().end();
    }
    
    void RemoveFromXlaCluster(NodeDef* node_def) {
      node_def->mutable_attr()->erase(kXlaClusterAttr);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/build_xla_ops_pass.cc

      MemoryTypeVector input_mtypes, output_mtypes;
      DeviceType device_type("");
      TF_RETURN_IF_ERROR(
          DeviceNameToDeviceType(n->assigned_device_name(), &device_type));
      TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(),
                                            device_type, n->def(), &input_mtypes,
                                            &output_mtypes));
      return output_mtypes;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc

        SmallVector<int64_t, 3> slice_size = {1, num_rows, num_cols};
        Type slice_result_type = RankedTensorType::get(slice_size, element_type);
        llvm::SmallVector<Type, 4> output_types(batch_size, slice_result_type);
        auto split_op = rewriter.create<TF::SplitOp>(loc, output_types,
                                                     split_dimension_op.getOutput(),
                                                     reshape_op.getOutput());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc

                  "max_percentile",
                  rewriter.getF32FloatAttr(
                      calib_opts_.calibration_parameters().max_percentile())),
          };
    
          SmallVector<Type, 4> output_types{
              value.getType(),
              RankedTensorType::get({}, rewriter.getF32Type()),
              RankedTensorType::get({}, rewriter.getF32Type()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 14.8K bytes
    - Viewed (0)
Back to top