HomeSort by relevance Sort by last modified time
    Searched refs:arg_sizes (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 41 std::vector<intptr_t> arg_sizes; local
43 arg_sizes.reserve(num_args);
47 arg_sizes.push_back(xla::ShapeUtil::ByteSizeOf(arg_shape, kPointerSize));
49 return std::move(arg_sizes);
156 TF_ASSIGN_OR_RETURN(std::vector<intptr_t> arg_sizes,
167 jit->arg_sizes_ = std::move(arg_sizes);
171 jit->static_data_.arg_sizes = jit->arg_sizes_.data();
xla_compiled_cpu_function.cc 36 static_data.arg_sizes, static_data.num_args, args_,
xla_compiled_cpu_function.h 64 const intptr_t* arg_sizes = nullptr; member in struct:tensorflow::XlaCompiledCpuFunction::StaticData
  /external/tensorflow/tensorflow/compiler/aot/
codegen.cc 100 // Fills in arg_sizes with the byte size of each positional arg.
102 std::vector<int64>* arg_sizes) {
105 arg_sizes->push_back(xla::ShapeUtil::ByteSizeOf(
297 std::vector<int64> arg_sizes; local
298 TF_RETURN_IF_ERROR(ComputeArgSizes(compile_result, &arg_sizes));
303 const std::vector<intptr_t> iarg(arg_sizes.begin(), arg_sizes.end());
403 static constexpr intptr_t kArgSizes[kNumArgs] = {{{ARG_SIZES}}};
413 data->arg_sizes = ArgSizes();
502 {"{{ARG_NUM}}", strings::StrCat(arg_sizes.size())}
    [all...]

Completed in 72 milliseconds