OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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