Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for tensor_b (0.26 sec)

  1. tensorflow/compiler/jit/deadness_analysis_test.cc

    TEST(DeadnessAnalysisTest, Recv) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output recv_a = ops::_Recv(root.WithOpName("recv_a"), DT_FLOAT, "tensor_a",
                                 "sender", 0, "receiver");
      Output recv_b = ops::_Recv(root.WithOpName("recv_b"), DT_FLOAT, "tensor_b",
                                 "sender", 0, "receiver");
      Output add = ops::Add(root.WithOpName("add"), recv_a, recv_b);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

    TEST(XlaCompilationTest, ClusterControlTrigger) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output recv_a = ops::_Recv(root.WithOpName("recv_a"), DT_BOOL, "tensor_a",
                                 "sender", 0, "receiver");
      Output recv_b = ops::_Recv(root.WithOpName("recv_b"), DT_BOOL, "tensor_b",
                                 "sender", 0, "receiver");
      Output const_a = ops::Const(root.WithOpName("const_a"), 42);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/deadness_analysis.cc

      // "tensor_id() is live and evaluates to true".
      //
      // If `must_be_true()` is false then this SymbolPredicate represents the
      // proposition "tensor_id() is live (and may evaluate to any value)"
      TensorId tensor_id() const { return tensor_id_; }
      bool must_be_true() const { return must_be_true_; }
    
     private:
      TensorId tensor_id_;
      bool must_be_true_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

      let summary = [{
    Op that copies host tensors to device with bounded dynamic shape support.
      }];
    
      let description = [{
    This op copies the padded tensor on cpu to TPU without the padded data. `tensors` 
    is a list of cpu tensors with padded data. `unpadded_sizes` is a list of shape
    tensors which describes unpadded size of each dimension for each cpu tensor. 
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

      // There should be 3 tensors: input, output, and BroadcastTo/shape.
      EXPECT_THAT(subgraph->tensors, SizeIs(3));
    
      // Input Tensor
      EXPECT_THAT(subgraph->tensors[0]->type, Eq(tensor_type_));
      EXPECT_THAT(subgraph->tensors[0]->name, Eq("input_1"));
      EXPECT_THAT(subgraph->tensors[0]->quantization->scale, SizeIs(1));
      EXPECT_THAT(subgraph->tensors[0]->quantization->zero_point, SizeIs(1));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

            correspond to.
          tags: Set of tags associated with the model.
          inputs: Input name -> input tensor mapping.
          outputs: Output name -> output tensor mapping.
          init_op: Op for initialization.
          assets_collection: Assets collection. This collection is a list of string
            tensors. Each tensor contains the asset file names.
        """
        v1_builder = builder.SavedModelBuilder(saved_model_path)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  7. .bazelrc

    build:cuda --@local_config_cuda//:enable_cuda
    
    # CUDA: This config refers to building CUDA op kernels with clang.
    build:cuda_clang --config=cuda
    # Enable TensorRT optimizations https://developer.nvidia.com/tensorrt
    build:cuda_clang --config=tensorrt
    build:cuda_clang --action_env=TF_CUDA_CLANG="1"
    build:cuda_clang --@local_config_cuda//:cuda_compiler=clang
    # Select supported compute capabilities (supported graphics cards).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 17:12:54 UTC 2024
    - 52.9K bytes
    - Viewed (0)
  8. tensorflow/c/c_api.h

    // - Directly referencing external tensors from the cond/body graphs (this is
    //   possible in the Python API)
    TF_CAPI_EXPORT extern TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs,
                                                     int ninputs,
                                                     TF_Status* status);
    
    // Builds the while loop specified by `params` and returns the output tensors of
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/tests/merge_control_flow.mlir

        %5 = "tf.IfRegion"(%0) ({
          %3 = "tf.B"(%4) : (tensor<f32>) -> (tensor<i32>)
          "tf.Yield"(%3) : (tensor<i32>) -> ()
          }, {
          %3 = "tf.D"(%4) : (tensor<f32>) -> (tensor<i32>)
          "tf.Yield"(%3) : (tensor<i32>) -> ()
         }) { is_stateless = true } : (tensor<i1>) -> (tensor<i32>)
        %6 = "tf.E"(%5) : (tensor<i32>) -> (tensor<f32>)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 63.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/tests/unroll-batch-matmul.mlir

      // CHECK: %[[LHS_SPLIT:.*]]:6 = "tf.Split"(%[[SPLITTING_AXIS]], %[[LHS_RESHAPED]]) : (tensor<i32>, tensor<6x4x5xf32>) -> (tensor<1x4x5xf32>, tensor<1x4x5xf32>, tensor<1x4x5xf32>, tensor<1x4x5xf32>, tensor<1x4x5xf32>, tensor<1x4x5xf32>)
      // CHECK: %[[LHS_1:.*]] = "tf.Reshape"(%[[LHS_SPLIT]]#0, %[[MATMUL_LHS_SHAPE]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> tensor<4x5xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 06 18:42:28 UTC 2023
    - 63.7K bytes
    - Viewed (0)
Back to top