Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ArgShapeAndDType (0.11 sec)

  1. tensorflow/compiler/jit/get_compiler_ir.cc

    BuildXlaCompilerArgumentFromTensorSpec(
        const FunctionBody* fbody, absl::Span<int const> must_be_constant_idxs,
        absl::Span<const Tensor* const> inputs,
        absl::Span<VariableInfo const> variable_args,
        absl::Span<const ArgShapeAndDType> flat_arg_shape_and_dtype) {
      TF_RET_CHECK(fbody != nullptr);
      auto& input_args = fbody->record->fdef().signature().input_arg();
      int input_arg_size = input_args.size();
      std::vector<XlaCompiler::Argument> args;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
Back to top