OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:shape_fn
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/framework/
variant_op_registry.cc
48
const VariantShapeFn&
shape_fn
) {
55
GetPersistentStringPiece(type_name),
shape_fn
));
62
UnaryVariantOpRegistry::VariantShapeFn*
shape_fn
=
local
64
if (
shape_fn
== nullptr) {
69
return (*
shape_fn
)(v, shape);
variant_op_registry.h
93
void RegisterShapeFn(const string& type_name, const VariantShapeFn&
shape_fn
);
327
const LocalVariantShapeFn&
shape_fn
) {
330
[type_name,
shape_fn
](const Variant& v, TensorShape* s) -> Status {
337
return
shape_fn
(*t, s);
variant_op_registry_test.cc
124
auto*
shape_fn
=
local
126
EXPECT_NE(
shape_fn
, nullptr);
131
Status s0 = (*
shape_fn
)(v, &shape);
137
TF_EXPECT_OK((*
shape_fn
)(v, &shape));
Completed in 599 milliseconds