Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for dyn_cast (0.05 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

    //
    // =============================================================================
    
    using tensorflow::AbstractFunction;
    using tensorflow::AbstractTensorHandle;
    using tensorflow::DataType;
    using tensorflow::dyn_cast;
    using tensorflow::OutputList;
    using tensorflow::Status;
    using tensorflow::unwrap;
    using tensorflow::wrap;
    using tensorflow::tracing::CreateTracingExecutionContext;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

          return errors::Internal("Expected 1 output but found ", num_outputs);
        }
        auto* t = dyn_cast<GraphTensor>(outputs[0]);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
        inputs_.push_back(t->output_);
        *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]);
        return absl::OkStatus();
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients_test.cc

                             /*raw_device_name=*/nullptr, &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
      if (isa<TracingOperation>(check_numerics_op.get())) {
        s = dyn_cast<TracingOperation>(check_numerics_op.get())
                ->SetOpName("check_numerics");
        ASSERT_EQ(errors::OK, s.code()) << s.message();
      }
      s = AddInput(check_numerics_op.get(), t.get(), &forward_op);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

      AbstractOperationPtr op(ctx->CreateOperation());
      TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr));
      if (isa<tracing::TracingOperation>(op.get())) {
        TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName(
            absl::StrCat("ZerosLike", ToId(t)).c_str()));
      }
      TF_RETURN_IF_ERROR(op->AddInput(t));
      int num_outputs = 1;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.cc

          // One of the inputs we're trying to pack is on a custom device. We'll let
          // the first custom device we see handle all of the packing.
          auto* custom_device_handle =
              tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>(
                  unwrapped_handle);
          tensorflow::ImmediateExecutionTensorHandle* result;
          status->status = custom_device_handle->device()->Pack(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 35.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

          tensorflow::unwrap(h);
      // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
      if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) {
        return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>(
                   unwrapped_handle)
            ->DevicePointer();
      }
      // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  7. tensorflow/c/c_api.cc

      tensorflow::TensorProto from_tensor_proto;
      status->status = BufferToMessage(from, &from_tensor_proto);
      if (!status->status.ok()) {
        return;
      }
      status->status =
          tensorflow::down_cast<tensorflow::TensorInterface*>(to->tensor)
              ->FromProto(from_tensor_proto);
    }
    // --------------------------------------------------------------------------
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top