OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:GetShape
(Results
1 - 10
of
10
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/kernels/
kernel_util_test.cc
51
std::vector<int>
GetShape
(TfLiteIntArray* dims) {
117
EXPECT_THAT(
GetShape
(output), ::testing::ElementsAre(1, 2));
124
EXPECT_THAT(
GetShape
(output), ::testing::ElementsAre(2));
134
EXPECT_THAT(
GetShape
(output), ::testing::ElementsAre(3, 1, 2));
141
EXPECT_THAT(
GetShape
(output), ::testing::ElementsAre(1, 2, 3, 4));
test_util.h
68
// m.BuildInterpreter({
GetShape
(a),
GetShape
(b)});
136
const std::vector<int>&
GetShape
(int id) { return tensor_data_.at(id).shape; }
/external/tensorflow/tensorflow/compiler/xla/client/
client.cc
362
StatusOr<Shape> Client::
GetShape
(const GlobalData& data) {
368
Status s = stub_->
GetShape
(&request, &response);
computation_builder.cc
224
StatusOr<std::unique_ptr<Shape>> ComputationBuilder::
GetShape
(
258
std::unique_ptr<Shape> actual_shape =
GetShape
(operand).ConsumeValueOrDie();
267
std::unique_ptr<Shape> lhs_shape =
GetShape
(lhs).ConsumeValueOrDie();
268
std::unique_ptr<Shape> rhs_shape =
GetShape
(rhs).ConsumeValueOrDie();
299
StatusOr<std::unique_ptr<Shape>> shape_status =
GetShape
(operand);
400
StatusOr<std::unique_ptr<Shape>> shape =
GetShape
(operand);
429
StatusOr<std::unique_ptr<Shape>> shape_or_status =
GetShape
(operand);
534
StatusOr<std::unique_ptr<Shape>> lhs_shape_or_status =
GetShape
(lhs);
635
StatusOr<std::unique_ptr<Shape>> lhs_shape_or_status =
GetShape
(lhs);
640
StatusOr<std::unique_ptr<Shape>> rhs_shape_or_status =
GetShape
(rhs)
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.h
183
const Shape&
GetShape
() const {
/external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc
307
std::unique_ptr<Shape> LocalComputationBuilder::
GetShape
(
309
return builder_.
GetShape
(operand).ConsumeValueOrDie();
xla_client.py
491
_unwrap_data_handle(operand), self.
GetShape
(operand),
639
def
GetShape
(self, operand):
640
return _wrap_shape(self._client.
GetShape
(_unwrap_data_handle(operand)))
734
padding, self.
GetShape
(operand).dimensions(),
909
padding, self.
GetShape
(operand).dimensions(), window_dimensions,
929
shape = Shape(self.
GetShape
(mu).np_dtype, dims)
949
shape = Shape(self.
GetShape
(a).np_dtype, dims)
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/
service.cc
[
all
...]
user_computation.cc
[
all
...]
/external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer.cc
[
all
...]
Completed in 239 milliseconds