OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:key_shape
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
lookup_util.h
60
TensorShape
key_shape
= keys_->shape();
local
61
if (!
key_shape
.IsSameSize(values_->shape())) {
65
key_shape
.DebugString(), " vs ", values_->shape().DebugString());
67
if (
key_shape
.num_elements() == 0) {
initializable_lookup_table.h
67
TensorShape
key_shape
() const final { return TensorShape(); }
lookup_table_op.cc
123
TensorShape
key_shape
() const final { return TensorShape(); }
252
TensorShape
key_shape
() const final { return TensorShape(); }
471
TensorShape
key_shape
= MaybeVectorizeShape(key_shape_);
variable
478
expected_value_shape.RemoveLastDims(
key_shape
.dims());
492
TensorShape
key_shape
() const override { return key_shape_; }
672
output_shape.RemoveLastDims(table->
key_shape
().dims());
[
all
...]
/external/tensorflow/tensorflow/core/framework/
lookup_interface.cc
25
if (!TensorShapeUtils::EndsWith(shape,
key_shape
())) {
28
key_shape
().DebugString());
52
for (int i = 0; i <
key_shape
().dims(); ++i) {
lookup_interface.h
90
virtual TensorShape
key_shape
() const = 0;
/external/tensorflow/tensorflow/contrib/lite/kernels/
hashtable_lookup_test.cc
36
std::initializer_list<int>
key_shape
,
45
BuildInterpreter({lookup_shape,
key_shape
, value_shape});
/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
predict_test.cc
38
std::initializer_list<int>
key_shape
,
52
BuildInterpreter({{input_signature_shape,
key_shape
, labelweight_shape,
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
sharded_mutable_dense_hashtable.py
87
key_shape
= keys.get_shape()
88
if
key_shape
.ndims > 1:
91
keys = array_ops.slice(keys, [0, 0], [
key_shape
[0].value, 1])
/frameworks/ml/nn/common/operations/
HashtableLookupTest.cpp
59
std::initializer_list<uint32_t>
key_shape
,
71
OperandType KeyTy(Type::TENSOR_INT32,
key_shape
);
Completed in 195 milliseconds