Searched defs:value_shape (Results 1 – 12 of 12) sorted by relevance
38 const TensorShape value_shape = ctx->InputShape("value"); in Compile() local
276 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local364 const TensorShape value_shape = ctx->InputShape(2); in Compile() local502 TensorShape value_shape = ctx->InputShape(1); in Compile() local
232 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local458 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local
216 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local
71 TensorShape value_shape() const final { return TensorShape(); } in value_shape() function
134 TensorShape value_shape() const override { return TensorShape(); } in value_shape() function in tensorflow::lookup::MutableHashTableOfScalars272 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableHashTableOfTensors536 TensorShape value_shape = MaybeVectorizeShape(value_shape_); in CheckKeyAndValueTensorsForImport() local558 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableDenseHashTable
37 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
38 std::initializer_list<int> value_shape) { in EmbeddingLookupSparseOpModel()
191 ShapeHandle value_shape = c->input(1); in CreateAssignShapeFn() local
677 ShapeHandle value_shape = c->input(2); in __anon4a7ac08d1402() local759 ShapeHandle value_shape; in __anon4a7ac08d1702() local
3522 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
4648 void Fill(const RuntimeShape& value_shape, const T* value_data, in Fill()