Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for axios (0.15 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

          } else {
            // Generalize differing axis lengths to "variable"/"unknown".
            for (int axis_index = 0; axis_index < combined_shape.dims();
                 ++axis_index) {
              int64_t axis_length = combined_shape.dim_size(axis_index);
              if (axis_length != component_shape.dim_size(axis_index)) {
                axis_length = -1;
              }
              TF_RETURN_IF_ERROR(
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  2. tensorflow/c/eager/parallel_device/parallel_device.cc

            // Non-parallel tensors from _EagerConst/tf.constant are implicitly
            // broadcast, i.e. set as the input to each parallel operation. This
            // allows code like "tf.constant(1.)" or "tf.reduce_sum(..., axis=1)"
            // (where the value starts on the host), without allowing other implicit
            // copies/broadcasts. Other implicit copies may be supported eventually,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.cc

    TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input,
                  TFE_TensorHandle* axis) {
      TF_Status* status = TF_NewStatus();
    
      TFE_Op* op = TFE_NewOp(ctx, "Min", status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, input, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, axis, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  4. tensorflow/c/eager/c_api_test.cc

      TFE_TensorHandle* axis = TestAxisTensorHandle(ctx);
      TFE_Op* minOp = MinOp(ctx, input, axis);
      TFE_TensorHandle* retvals[1] = {nullptr};
      int num_retvals = 1;
      TFE_Execute(minOp, &retvals[0], &num_retvals, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteOp(minOp);
      TFE_DeleteTensorHandle(input);
      TFE_DeleteTensorHandle(axis);
      ASSERT_EQ(1, num_retvals);
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  5. tensorflow/c/c_api_test.cc

      // Make a placeholder operation.
      TF_Operation* feed = Placeholder(graph, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      // Make a constant operation with the scalar "0", for axis.
      TF_Operation* one = ScalarConst(0, graph, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      // Create a session for this graph.
      CSession csession(graph, s, use_XLA);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top