Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 255 for cc_ops (0.13 sec)

  1. tensorflow/cc/BUILD

        hdrs = ["gradients/grad_helper.h"],
        visibility = ["//tensorflow:internal"],
        deps = [
            ":cc_ops",
            "//tensorflow/cc:ops",
            "//tensorflow/cc:scope",
        ],
    )
    
    cc_library(
        name = "linalg_grad",
        srcs = ["gradients/linalg_grad.cc"],
        deps = [
            ":cc_ops",
            ":cc_ops_internal",
            ":grad_helper",
            ":grad_op_registry",
            ":gradients",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc

    See the License for the specific language governing permissions and
    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
    
    #include <algorithm>
    #include <cstdint>
    #include <functional>
    #include <limits>
    #include <numeric>
    #include <string>
    #include <tuple>
    #include <type_traits>
    #include <utility>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

    // refreshes.
    //
    // This file includes another file, `tf_generated_ops.td`, which contains
    // all ops whose definitions are generated from TensorFlow codebase.
    // Changes made there are not respected.
    
    #ifndef TF_OPS
    #define TF_OPS
    
    include "tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td"
    include "tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td"
    include "mlir/Interfaces/CallInterfaces.td"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h

    Deqiang Chen <******@****.***> 1709591321 -0800
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 04 22:48:00 UTC 2024
    - 2.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_ops.td

    Deqiang Chen <******@****.***> 1716411971 -0700
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:35:32 UTC 2024
    - 6.7K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/cc_op_gen_main.cc

    #include "tensorflow/core/platform/env.h"
    #include "tensorflow/core/platform/init_main.h"
    #include "tensorflow/core/platform/types.h"
    #include "tsl/platform/status.h"
    
    namespace tensorflow {
    namespace cc_op {
    namespace {
    
    void PrintAllCCOps(const std::string& dot_h, const std::string& dot_cc,
                       bool include_internal,
                       const std::vector<string>& api_def_dirs) {
      OpList ops;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/dilated_conv.h

      // it's a constant. Basically, `paddings` tensor in `SpaceToBatch` and `crops`
      // tensor  in `BatchToSpace` must satisfy the following:
      //  paddings[i, 0] = base_paddings[i, 0].
      //  0 <= paddings[i, 1] - base_paddings[i, 1] < block_shape[i]
      // (input_shape[i] + paddings[i, 0] + paddings[i, 1]) % block_shape[i] == 0.
      //  crops[i, 0] = 0.
      //  crops[i, 1] = paddings[i, 1] - base_paddings[i, 1].
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

    from tensorflow.python.ops import array_ops_stack  # pylint: disable=g-direct-tensorflow-import
    from tensorflow.python.ops import io_ops
    from tensorflow.python.ops import lookup_ops
    from tensorflow.python.ops import math_ops
    from tensorflow.python.ops import nn_ops
    from tensorflow.python.ops import random_ops
    from tensorflow.python.ops import string_ops
    from tensorflow.python.ops import variables
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen_main.cc

    #include "tensorflow/core/platform/statusor.h"
    #include "tensorflow/core/platform/str_util.h"
    #include "tensorflow/core/platform/types.h"
    #include "tsl/platform/status.h"
    
    namespace tensorflow {
    namespace cc_op {
    namespace {
    
    void WriteAllFuzzers(string root_location, std::vector<string> api_def_dirs,
                         std::vector<string> op_names) {
      OpList ops;
      absl::StatusOr<ApiDefMap> api_def_map =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 10:53:49 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc

            return true;
          }
    
          // Check if the preceding op is a quantized same-scale op.
          if (llvm::isa<quantfork::StorageCastOp>(preceding_op)) {
            auto sc_op = llvm::cast<quantfork::StorageCastOp>(preceding_op);
            auto sc_arg_type = sc_op.getArg().getType().dyn_cast<TensorType>();
            if (sc_arg_type.getElementType().isInteger(8)) {
              return true;
            }
          }
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 23.6K bytes
    - Viewed (0)
Back to top