Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for reducer (0.18 sec)

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

          TFE_NewOp(context.get(), "AssertAndCollective", status.get()),
          TFE_DeleteOp);
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      std::unique_ptr<ParallelTensor> reduced_values =
          parallel_device.ScalarsFromSequence<float>({1.0, 2.0}, context.get(),
                                                     status.get());
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(),
                       final_op.length());
      TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0);
      TF_AddInput(reduce_desc, x);
      TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
      TF_Operation* operations[]{placeholder_op, reduce_op};
      TF_Output y{reduce_op, 0};
      const char* output_name = "y";
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  3. 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 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top