OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:shape0
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/contrib/graph_editor/
reroute.py
63
shape0
, shape1 = t0.get_shape(), t1.get_shape()
64
if not
shape0
.is_compatible_with(shape1):
65
raise ValueError("Shapes {} and {} are not compatible.".format(
shape0
,
/external/tensorflow/tensorflow/core/ops/
io_ops.cc
94
ShapeHandle
shape0
, shape1, shape2;
95
TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &
shape0
));
98
TF_RETURN_IF_ERROR(c->Merge(shape1, shape2, &
shape0
));
/external/tensorflow/tensorflow/compiler/tests/
concat_ops_test.py
259
for
shape0
in (), (2,):
260
axis = len(
shape0
)
264
x0 = np.random.randn(*(
shape0
+ (n0,) + shape1))
265
x1 = np.random.randn(*(
shape0
+ (n1,) + shape1))
/external/tensorflow/tensorflow/contrib/lite/toco/
tooling_util.h
107
// arrays d0 and d1. Without loss of generality, assume that
shape0
may have
108
// higher dimensionality (length(d0) >= length(d1)). Then
shape0
and shape1
113
bool ShapesAgreeUpToBroadcasting(const Shape&
shape0
, const Shape& shape1);
118
// arrays d0 and d1. Without loss of generality, assume that
shape0
may have
119
// higher dimensionality (length(d0) >= length(d1)). Then
shape0
and shape1
124
bool ShapesAgreeUpToExtending(const Shape&
shape0
, const Shape& shape1);
tooling_util.cc
553
bool ShapesAgreeUpToBroadcasting(const Shape&
shape0
, const Shape& shape1) {
554
CheckShapeDimensions(
shape0
);
557
const Shape* longer = &
shape0
;
559
if (shape1.dimensions_count() >
shape0
.dimensions_count()) {
561
shorter = &
shape0
;
581
bool ShapesAgreeUpToExtending(const Shape&
shape0
, const Shape& shape1) {
582
CheckShapeDimensions(
shape0
);
585
const Shape* longer = &
shape0
;
587
if (shape1.dimensions_count() >
shape0
.dimensions_count()) {
589
shorter = &
shape0
;
[
all
...]
/external/tensorflow/tensorflow/python/kernel_tests/
concat_op_test.py
362
for
shape0
in (), (2,):
363
axis = len(
shape0
)
367
x0 = np.random.randn(*(
shape0
+ (n0,) + shape1))
368
x1 = np.random.randn(*(
shape0
+ (n1,) + shape1))
/external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc
550
Shape
shape0
= ShapeUtil::MakeShape(S32, {9, 1, 4});
local
554
ShapeUtil::InsertedOrDeleted1SizedDimensions(
shape0
, shape1)));
556
ShapeUtil::InsertedOrDeleted1SizedDimensions(
shape0
, shape2)));
/external/tensorflow/tensorflow/compiler/xla/tests/
fusion_test.cc
785
Shape
shape0
= ShapeUtil::MakeShape(F32, {param0_dim0, param0_dim1});
local
[
all
...]
Completed in 1058 milliseconds